查看linux中存在的cuda版本,进入/usr/local目录
添加自己需要的cuda版本到环境变量,
vim ~/.bashrc
添加以下内容到最后:
#添加cuda路径
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export PATH=$PATH:/usr/local/cuda-10.1/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1
然后source一下使其生效source ~/.bashrc
最后查看cuda版本:nvcc -V
出现以下信息,则配置成功:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243