win10-cuda9.1+cuDNN7.1+pytorch1.1-GPU安装
pytorch1.1-GPU安装
对pytorch gpu版本的安装,稍微会复杂些,大概需要以下几个步骤:
(1)安装Nvidia Cuda;
(2)安装CuDNN;
(3)安装GPU版本pytorch;
(4)测试。
博主使用版本资源共享:链接:https://pan.baidu.com/s/18p4IdgfklEvQvo7FV09oVw
(提取码:336k )
具体步骤:
(1)安装Nvidia Cuda
首先检查看自己的电脑是否能安装cuda,右键->NVIDIA控制面板,
图一 点击NVID
原创
2020-07-06 12:13:21 ·
1602 阅读 ·
2 评论