第一步:安装显卡驱动
第二步:根据表中对应的显卡驱动找到适配的CUDA版本:
目前(2019年10月31日),很多深度学习库还不能很好的使用cuda10.1,最稳妥的版本选择10.0,
cuda10.0版本下载地址:
https://developer.nvidia.com/cuda-10.0-download-archive
我选择的runfile安装:
把.run文件下载到~/Download里之后,进入文件夹打开终端:
sudo chmod 777 cuda_10....run
sudo ./cuda_10....run
在选择
1. 是否安装驱动,选择no,因为已经安装了;
2. 是否创建/usr/local/cuda 的软连接时选no(n),后面环境变量直接写具体的版本,可以避免多版本cuda混淆;
3. 在阅读协议文档的时候连按 f ,快速跳到末尾,键入同意;
4. 默认安装地址/usr/local/cuda_10.0
报错:
Not enough space on parition mounted at /. Need ... bytes.
Disk space check has failed. Installation cannot continue.
说明整个Ubuntu系统文件位置不够用了,换一个分区吧,如果是显示/tmp空间不足,还可以使用下述命令重新安装来挽救:
sudo mkdir /opt/tmp
sudo cuda_10....run --tmpdir=/opt/tmp/
我把cuda安装到了另外的硬盘分区里,安装完成之后,需要执行以下步骤:
sudo gedit ~/.bashrc
添加以下路径:
export PATH=/media/maggie/Software/ZZUbuntu/cuda-10.0/bin:$PATH
export LD_LIBRARY_PATH=/media/maggie/Software/ZZUbuntu/cuda-10.0/lib64:$LD_LIBRARY_PATH
具体的路径根据你的安装路径来。
source ~/.bashrc
如果后面需要安装别的版本的cuda,那么就需要在多个cuda版本之间切换,这个时候需要先把./bashrc修改一下:
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
source生效。
用终端在上述地址建立cuda软链接:
sudo ln -s /media/maggie/Software/ZZUbuntu/cuda_10.0 /usr/local/cuda
nvcc -V检查是否成功。
切换之前,用下述命令先删掉原来的软链接:
rm -rf /usr/local/cuda