前提:已经安装了cuda8.0以及cuda9.0的Linux电脑
后期补充:这种删除快捷方式的方式不推荐,也可以直接用export命令,需要哪个cuda,就export哪个路径。
步骤:
1)将配置文件~/.bashrc中的cuda路径加上,不过加的是cuda8.0或者cuda9.0的快捷方式的路径。
export PATH="/usr/local/cuda/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"
export LIBRARY_PATH="/usr/local/cuda/lib64:$LIBRARY_PATH"
2)删除原有的cuda快捷方式,创建cuda8.0或者cuda9.0快捷方式,如这里,将cuda8.0链接到cuda的快捷方式。
rm -rf /usr/local/cuda
ln -s /usr/local/cuda-8.0 /usr/local/cuda
3)查看版本
nvcc --version
实时查看GPU使用:
watch -n 2 nvidia-smi
这些操作可能会导致不同版本不兼容、不匹配问题,建议使用docker,根本不用配置cuda,只需要安装好nvidia驱动就行了!!!!docker简单入门学习到迅速步入深度学习框架的使用,可以参考博客 。