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.csdn.net/Mind_programmonkey/article/details/99688839