赞
踩
切换至清华镜像源
cmd中输入:
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
添加Pytorch 清华源(清华源针对pytorch有单独的源)
cmd中输入:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
用conda创建新的虚拟环境
conda create -n 新环境名 python=版本号(新环境名自己随便取)
这里我用的:conda create -n pytorch1.4 python=3.7
切换至(激活)新的虚拟环境
conda activate 环境名
这里我的是:conda activate pytorch1.4
安装Pytorch1.4稳定版(2020-4-3号官网的稳定版是1.4)
到pytorch官网根据自己的情况进行选择:https://pytorch.org/
其中要选conda, CUDA版本的选择会要求电脑英伟达显卡驱动版本必须大于某个值,具体自行查。
将最下面的一行复制下来(末尾的 -c pytorch 不要复制,否则下载的源会默认是pytorch官方源,速度贼慢),我的是:
conda install pytorch torchvision cudatoolkit=10.1
验证
在(pytorch1.4) C:Users29530>后输入:python 回车 进入python环境,而后再输入import torch回车无异常表示pytorch安装成功, 然后输入: torch.cuda.is_available(),若返回 true,表示该pytorch可以使用GPU。
退出环境
输入:conda deactivate
其它
删除环境(不要乱删,这里只是说怎么做):conda remove -n 环境名
删除清华源并切换回原版源(不要乱删,这里只是说怎么做):
conda config --remove-key channels
作者:ApollooxX
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。