记录一下pytorch的安装过程。
我是参考的Pytorch(一) Pytorch 的安装_开开心心-CSDN博客_pytorch
虽然我的驱动版本很高,我还是选择了像上面链接那样做。
如果你下载cudnn有麻烦,可以试试下一个链接:
Index of /anaconda/pkgs/main/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
注:安装结束后将 ~/nvcc/bin(因为版本的不同可能在不同的地方) 目录添加到环境变量,我是通过搜索nvcc.exe找到的
测试cuddn需要用到的例子在:C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.1extrasdemo_suite
安装完cuddn之后,想着添加path环境变量:C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0libx64(具体路径看你自己的)
安装anaconda(就用不着安装python了,因为anaconda都包含了),可以在下面的链接里去找
Anaconda | The World's Most Popular Data Science PlatformAnaconda is the birthplace of Python data science. We are a movement of data scientists, data-driven enterprises, and open source communities.https://www.anaconda.com/
点击那个get start,我选的最新版本下载。
要想验证成功与否,装完成以后,在最近添加和anaconda包中都能找到anaconda prompt。
如果弹出黑窗口,显示base就证明成功了
安装pytorch,需要在pytorch官网上,做好选项如下:
之后,在上面anaconda命令行中就输入
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
就好了,等待。。。当然,中间需要输入yes or no
下载完成后我们再次输入pip list,查看到已经有torch的存在。
检验安装



