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