30系列显卡安装 pytorch 1.7.1 环境

1.安装torch

python的版本为3.8

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

2.安装依赖库

scipy==1.7.1 -i https://pypi.tuna.tsinghua.edu.cn/simple

numpy==1.21.2 -i https://pypi.tuna.tsinghua.edu.cn/simple

matplotlib==3.4.3 -i https://pypi.tuna.tsinghua.edu.cn/simple

opencv_python==4.5.3.56 -i https://pypi.tuna.tsinghua.edu.cn/simple

tqdm==4.62.2 -i https://pypi.tuna.tsinghua.edu.cn/simple

Pillow==8.3.2 -i https://pypi.tuna.tsinghua.edu.cn/simple

h5py==2.10.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

tensorboard==2.11.2 -i https://pypi.tuna.tsinghua.edu.cn/simple