win10安装pytorch简易教程
由于最近重装了电脑系统,环境需要重新配置,所以来这里做个笔记。
1、安装Anaconda
百度搜索清华镜像源,进入下载Anaconda安装文件,直接看我下载的, 点击exe文件一路next到底即可,中间注意勾选添加环境变量。 然后cmd进入命令行,输入:
conda create -n pytorch python=3.6.10
会在Anaconda安装目录下的envs中生成一个文件夹,其名为pytorch,其中的python环境是python 3.6.10,这个你可以随便取,只要python官网有就行。 然后在安装之前可以切换conda镜像源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --set show_channel_urls yes
然后删除默认镜像源,必须删了!!!
conda config --remove channels defaults
2、安装pytorch
首先进入刚刚创建好的虚拟环境,
conda activate pytorch
然后点击进入,选择相应版本,这里我选择的是windows、cuda10.2、python版本: 直接复制其中的命令到终端即可进行安装。
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch conda install pytorch torchvision cudatoolkit=10.2
注意,务必去除-c pytorch这串字符,要不然还是用的pytorch官网的镜像源,换源就毫无用处。
接下来静静等待安装完成。 等待片刻,如果安装成功,那么进入pytorch虚拟环境,输入python,回车。 然后输入
import torch # 不报错 import torchvision # 不报错 torch.cuda.is_available() # 返回值应该是True
3、小技巧
如果安装失败,没关系,深吸口气,博主这里很贴心,只需要完成以下几步,一般可以成功装上。 第一步: 删除所创建的虚拟环境
conda remove -n pytorch --all
第二步: 卸载anaconda 第三步: 重启电脑 第四步: 打开本博客,重新来一遍。
预祝大家都能装成功!