快捷搜索: 王者荣耀 脱发

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 第三步: 重启电脑 第四步: 打开本博客,重新来一遍。

预祝大家都能装成功!

经验分享 程序员 微信小程序 职场和发展