nvidia-smi后这里可以看到cuda的版本是11.1但是实际上我们的不一定是11.1的版本,使用的版本需要通过nvcc -V来查看具体的使用的cuda版本号。
nvcc -V显示如下(这个图是截图别人的):我们可以看到这里的版本是9.0
有时候服务器没有nvcc -V这个命令找不到,我们可以通过 ls /usr/local看下服务器安装了几个cuda,一般服务器上都可以看到有cuda的版本,如果只有一个cuda就可以知道当前服务器的本版,如果是有几个本版,我们可以再通过echo $PATH来看指定的cuda版本是哪一个。修改服务器指定的cuda版本,临时指定版本可以用export PATH='cuda的位置' $PATH,然后再echo一下看啊可能现在的版本是哪一个。如果要永久PATH可以看看vi ~/.bashrc 内的PATH,如果这里没有的话,可以去vi /etc/profile
总结:nvidia-smi的版本可以高于nvcc -V看到的版本,如果nvcc -V看到的版本高于则会出错,需要安装相应的驱动drive 这种情况需要参考官网按照的要求。
注:
export CUDA_VISIBLE_DEIVES=2导入其他卡