安装远程linux CUDA11.3对应的pytorch

conda create -n 【env_name】python=3.8

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113  -i https://pypi.tuna.tsinghua.edu.cn/simple

其他版本可参考：https://pytorch.org/get-started/previous-versions/

.bashrc文件中添加：

# <<< conda initialize <<<
export PATH=/home/xxxxxxxxxxxxxxxxxxxxxxxxxxx/anaconda3/bin:$PATH
export CUDA_ROOT=/usr/local/cuda-11.3${CUDA_ROOT:+:${CUDA_ROOT}}
export PATH=/usr/local/cuda-11.3/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

(xxxxxxxx)修改为用户名

修改后执行

source ~/.bashrc