安装Pytorch的gpu版本时如何设置系统cuda路径

设置系统cuda路径。在.bashrc里面添加如下路径

export PATH=/usr/local/cuda-11.8/bin:$PATH
export CUDA_PATH=/usr/local/cuda-11.8
export CUDA_LIBDIR=/usr/local/cuda-11.8
export CUDA_LIBRARY_PATH=/usr/local/cuda-11.8
export LD_LIBRARY_PATH=/usr/local/cuda-11.8

系统cuda版本与pytorch版本关联，不兼容的cuda会导致无法正确安装pytorch-gpu。
修改完路径后记得source ~/.bashrc