windows10安装Anaconda,cuda和GPU版Pytorch (简单详细)
1.Anaconda
安装anaconda
link
2.Cuda
如果要使用GPU版本的Pytorch则必须安装Cuda
link
![在这里插入图片描述](https://img-blog.csdnimg.cn/20201206000702697.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nL
3.Pytorch
在官网找到适合自己的Pytorch版本
link
这时候打开anaconda prompt, 创建一个虚拟环境,名字随便起,我这里叫"torch_py38" (提醒自己用的是python3.8).
你在创建环境的时候也要注意自己使用的是什么版本的python,python版本,cuda版本和anaconda版本一定都要保持一致要不很容易有奇怪的bug…OvO’’
然后我们进入这个环境, 可以看到 base -> torch_py38
然后输入上面在pytorch官网找到的命令
我的版本是这个
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
装好之后测试一下:
如果第一个命令得到True就大功告成了。
如果第一个命令false但是底下的测试成功那么代表pytorch还是没有使用你的GPU,这时候就得倒回去检查是不是前面哪一步做漏了。
也有可能是出现了其他的bug,这里举例几个:
- 没有更新显卡驱动
显卡驱动可以到NVDIA官网自己更新下载:link
选择自己的型号就好 - python 版本不对,这边建议简单粗暴直接删除刚才创建的虚拟环境然后从创建虚拟环境那一步开始使用正确的python版本从做一次
- pytorch 版本不对,在刚才的虚拟环境里,输入 conda uninstall pytorch, 然后找到正确的版本命令重装pytorch。记得一定要先把之前的版本给卸载掉要不新的版本安装会出很多问题 (0.0
以上就是在windows 10 系统下安装 GPU版的Pytorch的全部过程了
(文中在最开始省略了安装python的步骤,
但是笔者认为没有必要讲解这一步,
如果您因为没做这一步而没法成功安装…)