使用pytorch进行深度学习一共要安装四样东西:
1.CUDA+CUDNN
2.pycharm
3.Anoconda
4.pytorch
NOTE:
1.整个安装过程我是按照视频来进行的。其中要注意CUDA的版本和Anoconda的python版本和pytorch的版本,一定要对应起来。
https://www.bilibili.com/video/BV1k7411j7S9?p=1
2.唯一的区别是视频没有进行第一步的CUDA的安装,自己先安装CUDA后可以按照视频安装。
一、CUDA+CUDNN的下载安装与环境变量设置
1.CUDA下载安装
网址:https://developer.nvidia.com/cuda-toolkit-archive
2.CUDA具体的安装步骤参考帖子与环境变量设置
网址:https://blog.csdn.net/caomin1hao/article/details/103672614
3、CUDNN的下载与安装
1.下载网址CUDNN下载链接 注意:下载的版本要跟你CUDA的版本对应。
2.下载CUDNN需要注册,简单注册即可。
3.将下载后的解压包解压出来,将解压出来的三个文件夹文件全部复制粘贴到安装的CUDA的目录下。
复制
粘贴:
粘贴的路径为安装CUDA的路径 我的CUDA是安装在 :C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1中
二、安装pycharm
网上有许多的安装步骤,这个省略
三、安装Anoconda
1.下载Anoconda,下载链接为Anoconda下载链接
有一点特别要注意的是:下载的时候看其python版本,因为这个是python版本一定要跟pytorch的版本一致。这个非常重要。这个非常重要。这个非常重要。
2. 安装是时候到了这一步要勾选第一个将其自动增加到环境变量。其他责默认即可。
四、安装pytorch
这个是最重要的,想要安装pytorch成功则先必须先将前面三个安装。
1.下载pytorch,下载的版本跟CUDA和Anoconda的python版本息息相关。
如上图CUDA版本和noconda的python都有说明,具体的下载安装过程可以参考视频:参考视频
总结
安装务必小心,作为新手多试几次,不要害怕失败。