问题描述:
最近在linux 服务器上,由于root下安装的cuda版本较老,需要在自己的家目录下安装新版本的cuda,按照网上的博客教程完成安装后,输入nvcc -V
发现输出仍为旧版本的cuda
主要原因:
nvcc -V
输出不正确是因为环境变量的配置问题,需要将新安装的cuda目录路径设置在开头而非末尾,使得系统在寻找cuda路径时先找到新版本的cuda,如下所示,/home/gwc/cuda-11.3
为我新安装的cuda目录,$CUDA_HOME
则为原来的cuda目录:
export CUDA_HOME=/home/gwc/cuda-11.3:$CUDA_HOME
export PATH=/home/gwc/cuda-11.3/bin:$PATH
export LD_LIBRARY_PATH=/home/gwc/cuda-11.3/lib64:$LD_LIBRARY_PATH
而不是以下方式,以下方式系统仍使用旧cuda,因为新cuda路径放在了最后,旧cuda的路径仍旧在前:
export CUDA_HOME=$CUDA_HOME:/home/gwc/cuda-11.3
export PATH=$PATH:/home/gwc/cuda-11.3/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/gwc/cuda-11.3/lib64
设置好cuda环境变量后,输入nvcc -V,输出变为新版本,如下所示:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Sun_Mar_21_19:15:46_PDT_2021
Cuda compilation tools, release 11.3, V11.3.58
Build cuda_11.3.r11.3/compiler.29745058_0