原本在服务器上个人账号下安装的是cuda9.2版本(NVIDA默认的最新版本),但是运行代码时报错。
这是跟driver版本对不上的原因。查看了一下driver版本是384.111
接着找到了这个网址https://stackoverflow.com/questions/30820513/what-is-version-of-cuda-for-nvidia-304-125/30820690#30820690,里面是driver对应的cuda版本,如下:
所以集群上的driver对应的cuda版本应该是9.0。但是drive是在集群管理员权限下的,我没有修改权限,那就只好改cuda咯。
首先进入到cuda的安装目录,然后卸载cuda,命令如下