新手第一次安装,记录自己安装经过。
1.安装anaconda
直接在anaconda官网上选择相应版本安装包安装即可。anaconda自带python,为避免和已有python冲突,可将已有版本删除,之后将anaconda中python安装目录加入环境变量。
Scripts也要加入环境变量
2.安装pytorch
打开anaconda prompt
输入命令:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --set show_channel_urls yes conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda install pytorch torchvision cudatoolkit=10.0
等待出现successful即可,速度挺快的。
3.安装CUDA和cudnn
CUDA我之前装过了,直接就跳过了。不过安装时注意选择自定义安装,之后一路下一步就好。安装后的位置如图。
cudnn去官网下载,下载时要完成一个简单的注册才能下载。注意下载时选择和自己CUDA匹配的版本。下载下来是一个压缩包,只需要把每个文件夹的内容复制到CUDA中同名文件夹下即可完成安装,cudnn中单独的那个文件复制到上图文件夹中即可。
4.创建虚拟环境
首先在anaconda prompt中切换到anaconda中的envs的文件夹下,之后输入
conda create -n pytorch(虚拟环境名) python=3.9
完成创建虚拟环境,之后输入
activate pytorch
即可激活环境。
5.安装torch和torchvision
在命令提示符中先pip安装wheel包,之后直接pip先安装torch,再安装torchvision。可能是使用了梯子的原因,安装速度不慢。
完成安装。
标签:Windows,python,pytorch,conda,anaconda,--,安装 From: https://www.cnblogs.com/y0hv2y/p/17028958.html