torch+ CUDA gpu 并行计算环境配置
需要注意 GPU算力值 cuda版本 显卡驱动版本 torch版本 之间的匹配
torch版本可以根据cuda版本来
先根据驱动版本和显卡算力(显卡算力自己查)选择合适cuda版本
我的显卡GT730 算力3.5 驱动最高支持版本11.4 选择cuda版本8.0
然后安装下载安装cuda 并且配置环境cuda环境变量
最后根据cuda版本选择pytorch版本,并安装
以上是安装torch+cuda的步骤 具体实例安装请参考:https://blog.youkuaiyun.com/Mind_programmonkey/article/details/99688839
本文详细介绍了如何根据GPU算力和驱动版本选择合适的CUDA版本,以GT730显卡为例,选择了CUDA 8.0。接着,讲述了安装CUDA并设置环境变量的步骤,最后强调了根据CUDA版本选择对应PyTorch版本进行安装的重要性。提供了一个具体的安装教程链接以供参考。
2万+





