Pytorch安装
一、安装anaconda
网址:https://www.anaconda.com/products/individual
网页翻到最下面,选择下载Windows 32位/64位版本,傻瓜式安装,记得勾选【Add to Path】 可以将anaconda的环境安装到系统中去
二、Anaconda创建虚拟环境
1 、从开始菜单中打开Anaconda Navigator,单机进入Environments选项卡,如下图所示
2、在Environments选项卡中,单击左下方的Create按钮,创建虚拟环境,如下图所示
此时在Environments选项卡中,已经出现刚刚创建的虚拟环境 【注】才创建的虚拟环境只包含最基本的库
三、安装Pycharm,创建工程
网址:https://www.jetbrains.com/pycharm/
1、单击download进入下载页面
此时,有两种版本选择,嫌麻烦就用社区版,但是功能较少,专业版需要自己找破解,或者用校园邮箱申请一年免费,一年后再次审查通过还可以继续用
2、打开PyCharm,单击New project
3、PyCharm创建Conda工程
4、在Anaconda安装文件夹下找到envs文件夹,里面存储的都是你创建的虚拟环境,然后选择刚刚创建的虚拟环境文件下的python.exe,点击OK
5、创建完成后自动进入如下界面:
四、CUDA安装
1、查看CUDA支持版本
打开NVIDIA控制面板,按照图示顺序查看CUDA所支持的最高版本
2、下载CUDA
打开网址:https://developer.nvidia.com/cuda-toolkit-archive,选择适合自己电脑的cuda版本,比刚才查看的所支持的CUDA 版本低都行
此时要根据自己的系统进行选择
3、下载cuDNN
打开网址:https://developer.nvidia.com/cudnn,点击Download cuDNN
注意,cuDNN下载需要NVIDIA账户,没有则先注册,此步需要耐心,贼卡
4、登陆之后,点击Download cuDNN进入的应该是如下界面,先把问卷填了,才能进入下载界面
此时终于进入下载界面了,勾选I agree…,然后按照下图操作
5、选择适合自己的操作系统版本下载
6、安装CUDA
双击安装包开始安装,此时让你选择的路径并不是安装路径,而是暂时解压出来的文件所需要存放的地方,安装完成后会删除,只要保证选择空间足够大的地方即可,安装路径是默认C盘里,不能修改,此处可以不更改,点击OK即可
验证CUDA是否安装成功,打开cmd,输入 nvcc –V,如下返回则说明安装成功
解压压缩包,压缩包内文件如下:
7、安装cuDNN
打开cuda安装文件夹,默认的强制安装路径为:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\版本号
打开安装路径如下:
将解压的cuDNN压缩包内的三个文件夹复制到CUDA安装目录下:
验证cuDNN是否安装完成,打开cmd,输入
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\extras\demo_suite
然后执行命令:
bandwidthTest.exe
然后再执行命令:
deviceQuery.exe
8、下载PyTorch安装文件
打开网址:https://download.pytorch.org/whl/torch_stable.html,界面如下:
每个文件命名都是有规律的:
使用快捷键Ctrl+F,搜索我们需要安装的版本文件下载
五、安装pytorch
1、打开PyCharm,进入刚刚创建的工程里,打开Terminal窗口:
2、在PyCharm的Terminal窗口中键入cd 命令,切换目录到PyTorch安装包所在的地址,然后分别执行pip install命令安装两个文件,先安装torch,再安装torchvision,期间会一起下载所需要的依赖包
3、现在验证是否安装成功,在main.py里写入以下代码并运行:
以上方式针对的是windows10操作系统,大家可以根据大家各自的操作系统选择,安装cuda或者不安装cuda