1.下载anaconda3(清华源)
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ 点击链接选择合适版本的anaconda进行安装
2.anaconda安装pytorch
安装好anaconda后点击电脑左下微软图标找到anaconda所在文件,点击下图所示位置anaconda Prompt进入命令行
在命令行执行如下操作:
创建一个torch环境:输入:conda create -n 环境名称 python=3.6
如:conda create -n torch python=3.6
激活所建环境:输入:activate 环境名称(如果要删除环境输入:conda remove -n your_env_name(虚拟环境名称) --all)
如:activate torch
激活之后左边名称发生变化
然后Conda 修改镜像源
查看目前已有的channel:
conda config --show channels(注意show前是两个短横线)
添加新的channel:(使用清华源)https://mirror.tuna.tsinghua.edu.cn/help/anaconda/
复制下面几行代码输入到命令行
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
现在通道已经换成清华源:
在安装pytorch时,本人使用的cuda版本为10.2
进入pytorch官网https://pytorch.org/get-started/locally/
将图中命令复制到命令行
conda install pytorch torchvision cudatoolkit=10.2(用镜像不能末位加上-c pytorch)
安装好之后如图所示在命令行输入python,回车键
输入import回车
输入print(torch.cuda.is_available())
结果显示为True,torch安装成功。
3.pycharm配置pytorch环境
创建新项目,点击图上位置
新建一个.py文件输入以下代码,
结果为