深度学习环境配置
由于最近重装了电脑系统,环境需要重新配置,所以来这里做个笔记。
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
然后点击进入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
第三步:
重启电脑
第四步:
打开本博客,重新来一遍。
预祝大家都能装成功!