下载安装pytorch

下载安装pytorch

在：
https://download.pytorch.org/whl/torch_stable.html

根据服务器CUDA的版本：

nvidia-smi

找到对应版本的pytorch：

下载之后，安装：

pip install cu113/torch-1.12.1%2Bcu113-cp38-cp38-linux_x86_64.whl

搞定！