使用anaconda3建立pytorch虚拟环境(Windows/Linux)--个人虚拟环境

虚拟环境：包含一系列功能的安装包

由于在anaconda默认环境下使用pip install pytorch总是报安装错误，所以考虑新建一个pytorch的虚拟环境。

1、创建虚拟环境

打开Anaconda Prompt
输入命令行：
conda create -n pytorch python=3.6

注意：如果是在Linux中需要先添加conda环境：

vim ~/.bashrc

#输入一下代码,注意export PATH

_conda_setup="$(CONDA_REPORT_ERRORS=false '/home/ddddd/anaconda3/bin/conda' shell.bash hook 2> /dev/null)"
     if [ $? -eq 0 ]; then
        \eval "$__conda_setup"
     else
         if [ -f "/home/ddddd/anaconda3/etc/profile.d/conda.sh" ]; then
            . "/home/ddddd/anaconda3/etc/profile.d/conda.sh"
             CONDA_CHANGEPS1=false conda activate base
         else
             \export PATH="/home/自定义路径/anaconda3/bin:$PATH"
         fi
     fi

#刷新配置文件
source ~/.bashrc

#测试conda 是否配置成功
conda list

创建结果如图所示：

注意：创建完虚拟环境以及安装完成之后一定要在pycharm中切换虚拟环境，虚拟环境都在anaconda3/envs/目录下显示。

找到pytorch下单python.exe即可

如图：

 

2、添加清华镜像（快速下载虚拟环境所需的安装包）

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --set show_channel_urls yes

3、激活虚拟环境

activate pytorch(conda activate pytorch)

#取消激活--conda deactivate(conda deactivate pytorch)

4、配置CPU/GPU资源

# for CPU only packages
conda install -c peterjc123 pytorch

# for Windows 10 and Windows Server 2016, CUDA 8
conda install -c peterjc123 pytorch cuda80

# for Windows 10 and Windows Server 2016, CUDA 9
conda install -c peterjc123 pytorch cuda90

# for Windows 7/8/8.1 and Windows Server 2008/2012, CUDA 8
conda install -c peterjc123 pytorch_legacy cuda80

5、torchvision的安装

本地安装-->
下载安装包-->https://pypi.python.org/pypi/torchvision/0.1.8
输入命令行：pip install 安装包名字.whl

6、测试

7、测试cuda和cuDNN