Win10+Anaconda创建新环境安装pytorch
1、打开Anaconda Prompt
2、输入下载地址(参考来源https://github.com/dragen1860/Deep-Learning-with-PyTorch-Tutorials)
# 配置国内源,方便安装Numpy,Matplotlib等
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/
# 配置国内源,安装PyTorch用
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# 显示源地址
conda config --set show_channel_urls yes
将上述地址单个输入到Anaconda Prompt中,如下图:
3、创建一个新环境并**:
# 创建一个名为pytorch的环境,python版本是3.7
conda create -n pytorch python=3.7
# **名为pytorch环境
conda activate pytorch
这时看到Anaconda界面会多出一个新的环境(你创建的环境),且在anaconda3\envs文件夹下多了一个pytorch的文件夹:
4、安装pytorch(下面输入语句详见官网,下面这条语句作为参考)
# 安装PyTorch,要使用国内源请去掉-c pytorch这个参数!!
conda install pytorch torchvision cudatoolkit=10.0
5、验证安装是否成功
在当前环境下输入:
# 进入python
pythonimport torch
import torchvision
# 打印版本
print(torch.__version__)# 退出python
exit()
# 退出pytorch
conda deactivate
注:使用activate pytorch进入该环境下,已经在该环境下不需要使用。