ubuntu 下3080Ti 环境配置记录

ubuntu 下3080Ti 环境配置记录

1.torch1.7.0

//创建环境
conda create -n ENV1 python=3.7
#创建完成后进入环境
conda activate ENV1
#安装pytorch
pip install torch==1.7.0+cu110 torchvision==0.8.1+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
#pip超时，加上timeout的时间
pip install --default-timeout=1000 torch==torch1.7.0+cu110 torchvision==0.8.1+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
//conda 下安装cuda
conda install cudatoolkit=11.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
//conda 下安装cudnn
conda install cudnn=8.0.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/
//不写版本会自动选版本与cuda版本相匹配
conda install cudnn

统一存放在envs/ENV1/lib or include 文件夹下：

命令行查看虚拟环境cuda版本，具体对应可以pytorch官网

//命令行查看虚拟环境cuda版本
conda list | grep cuda

命令行查看虚拟环境cuda版本，具体对应可以pytorch官网

1.torch1.7.1

//创建环境
conda create -n ENV2 python=3.6
#创建完成后进入环境
conda activate ENV2
#安装pytorch
pip install torch==torch1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
#pip超时，加上timeout的时间
pip install --default-timeout=1000 torch==torch1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

1.torch1.10.1

//创建环境
conda create -n ENV3 python=3.7
#创建完成后进入环境
conda activate ENV3
#安装pytorch
pip install torch==torch1.10.1+cu102  torchvision==0.11.2+rocm4.2 -f https://download.pytorch.org/whl/torch_stable.html
#pip超时，加上timeout的时间
pip install --default-timeout=1000 torch==torch1.10.1+cu102  torchvision==0.11.2+rocm4.2 -f https://download.pytorch.org/whl/torch_stable.html