Windows安装pytorch
新手第一次安装,记录自己安装经过。
1.安装anaconda
直接在anaconda官网上选择相应版本安装包安装即可。anaconda自带python,为避免和已有python冲突,可将已有版本删除。anaconda安装时记得一定要勾选上添加环境变量这一选项。成功安装后可在命令行中输入conda --version进行测试。
2.安装CUDA和cudnn
CUDA我之前装过了,直接就跳过了。不过安装时注意选择自定义安装,之后一路下一步就好。安装后的位置如图。
cudnn去官网下载,下载时要完成一个简单的注册才能下载。注意下载时选择和自己CUDA匹配的版本。下载下来是一个压缩包,只需要把每个文件夹的内容复制到CUDA中同名文件夹下即可完成安装,cudnn中单独的那个文件复制到上图文件夹中即可。
3.创建虚拟环境
打开anaconda prompt或者命令行,输入
conda create -n pytorchlearn(虚拟环境名) python=3.9
完成创建虚拟环境,之后输入
activate pytorchlearn
即可激活环境。
4.安装torch和torchvision
在命令提示符中先pip安装wheel包,之后直接pip先安装torch,再安装torchvision。可能是使用了梯子的原因,安装速度不慢。
完成安装。