DJTU 安装torch

1.当拉取pytorch镜像时没有pip的问题时可使用以下方式

apt update
apt install python3-pip

2.安装pytorch

清华源镜像 如遇到torch版本不符可修改版本号

pip install torch==1.10.0 torchvision==0.11.1 -i https://pypi.tuna.tsinghua.edu.cn/simple

阿里源

pip install torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 -i https://mirrors.aliyun.com/pypi/simple

官方源（很慢）

pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html

3.检验是否安装成功（检验是否能用GPU帮助训练）

返回True则表示安装成功。False表示失败，可能是 CUDA 安装或硬件兼容问题。

python3

import torch

torch.cuda.is_available() 

返回True就可以继续安装其他包了