ubuntu18.04 conda虚拟环境配置cuda11.0、pytorch1.7.0

1. nvidia官网下载cuda并安装

下载地址，下载cuda11.0版本的.run文件。

进入安装文件所在文件夹，cuda_10.1.105_418.39_linux.run是我们下载的cuda安装文件
$ sudo sh cuda_11.0.2_450.51.05_linux.run  

接下来耐心等一小会，会出现命令行安装界面，选择“continue”、“accept”、回车取消勾选下载驱动的选项，最后选择安装。注意，因为我系统中本来已经安装了cuda11.3，所以系统cuda本已软连接到了11.3上，这次装cuda11.0是为了conda虚拟环境使用，所以安装的最后要选择不修改环境变量。

这时候进入/usr/local目录，输入’stat cuda’命令也会发现输出结果是‘cuda-11.3’。

2. conda虚拟环境下载pytorch1.7.0

首先创建名为pytorch-1.7的虚拟环境
$ conda create -n pytorch-1.7 python=3.6
进入虚拟环境
$ source activate pytroch-1.7
安装pytorch 1.7以及cudatoolkit 11.0
$ conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0

3. 为conda虚拟环境设置CUDA版本

这个部分参考文章

输出虚拟环境的路径：/home/usrname/.conda/envs/pytorch-1.7
$ echo ${CONDA_PREFIX}

建立激活环境下的文件夹
$ mkdir -p /home/username/.conda/envs/pytorch/etc/conda/activate.d
写入脚本
$ vi /home/username/.conda/envs/pytorch/etc/conda/activate.d/activate.sh

进入脚本内容如下：

#!/bin/bash

ORIGINAL_CUDA_HOME=$CUDA_HOME
ORIGINAL_LD_LIBRARY_PATH=$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda-11.0
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH

#建立退出环境下的文件夹
mkdir -p /home/username/anaconda3/envs/pytorch/etc/conda/deactivate.d
#写入脚本
vi /home/username/anaconda3/envs/pytorch/etc/conda/deactivate.d/deactivate.sh

退出脚本内容如下：

#!/bin/bash

export CUDA_HOME=$ORIGINAL_CUDA_HOME
export LD_LIBRARY_PATH=$ORIGINAL_LD_LIBRARY_PATH
unset ORIGINAL_CUDA_HOME
unset ORIGINAL_LD_LIBRARY_PATH

4. 测试

首先退出虚拟环境，再进入
$ source deactivate
$ conda activate pytorch-1.7

查看环境变量CUDA_HOME，结果为/usr/local/cuda-11.0，说明虚拟环境所用cuda版本是11.0的
$ echo $CUDA_HOME

再退出虚拟环境
$ source deactivate
$ cd /usr/local
查看系统所用cuda，发现还是11.3
$ stat cuda 

到这里，为虚拟环境指定cuda版本就成功了。