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

歪比巴卜(｢･ω･)｢

已于 2023-12-21 11:48:39 修改

阅读量2.2k

点赞数 6

文章标签： pytorch linux 深度学习

于 2023-03-26 09:25:22 首次发布

本文链接：https://blog.csdn.net/qq_55335453/article/details/129775864

版权

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)修改为用户名

修改后执行