1 从清华源下载anaconda镜像,我安装的是下面这个
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/?C=M&O=D
2 打开文件所在的终端,输入:
bash ~/Downloads/Anaconda3-2020.02-Linux-x86_64.sh
3 输入:
export PATH=/home/xjgao/anaconda3/bin:$PATH
再输入:
source ~/.bashrc
这时候应该进入了base,在base中创建新环境,并进入:
conda create -n torch1.6 python=3.7
conda activate torch1.6
这时候就进入了新创建的这个环境。接下来在这个环境中安装cuda,这个要从pytorch官网下载:
https://pytorch.org/get-started/locally/
选准要装的版本,复制下面的命令到终端运行
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
这时候要下载非常多的东西,非常慢,慢慢等。
到这时就已经安装完毕,代码放在这个环境下运行,需要什么包也安装在这个环境下。