torch+cuda gpu并行计算

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

你可能感兴趣的:(pytorch,深度学习,人工智能)