背景
依据不同的torch版本,可能需要不同的cuda版本,我们可以同时安装多个cuda版本,然后是使用不同的cuda版本时在配置环境中进行修改。
下载链接
直接下载bash 可以运行的文件,然后按照要求回车就可以。
https://developer.nvidia.com/cuda-toolkit-archive
环境配置
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64/:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-9.0/bin:$PATH
export CUDA_HOME=/usr/local/cuda-9.0
查看版本
安装好之后,退出命令窗重新进入,使用命令:
nvcc -V