PyCuda配置(一)Ubuntu16.04 + GeForce 940 配置CUDA

本文是在配置PyCuda包的第一个步骤,简单介绍了如何在ubuntu 16.04 环境下配置Cuda。

本文分为两部分,第一部分介绍安装流程,第二部分介绍常见问题与解决方法

 

一、 配置Cuda

配置Cuda的过程简单来说就是下载Cuda包,然后安装即可。但是在安装的过程中可能出现比较多的问题,最严重的会导致系统卡在登录界面循环登录。

0 预先检查

首先要检查自己的显卡是否能用Cuda,请参考以下博文中的第二步。

https://blog.csdn.net/qlulibin/article/details/78714596

一般来说是没问题的,保险起见请检查一下。上文中博客的方法也是官网给出的方法,只是翻译成了中文。

1 确定Cuda版本

查询本机NVIDIA显卡驱动对应的Cuda版本:https://blog.csdn.net/BigData_Mining/article/details/103698420

确定本机驱动可以用命令行 nvidia-smi,我的驱动型号为 384.130,因此下载了Cuda8.0 GA2,保险起见没有下载Cuda9.0。

2 下载Cuda

在官网:https://developer.nvidia.com/cuda-toolkit-archive 下载对应的Cuda包,选择步骤1中的版本。

3 安装

打开下载后文件所在文件夹,按照官网的提示在命令行输入命令

sudo sh cuda_8.0.61_375.26_linux.run

除了是否安装驱动那里选择否,其他都选择是(accept或y)。

安装后会出现成功界面,再按照提示于~/.bashrc中添加路径,用vim打开~/.bashrc,然后在末尾输入下述指令,请注意核对命令行中的路径。 

export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

 4 验证

新开一个终端,命令行中输入

nvcc --version

如果成功显示则已成功,否则要参考问题2,先卸载cuda包,再重新安装。另外,最好再进行一下链接博文中的“五、测试CUDA的Sample例子”来进行测试。

更详细的问题可以参考https://blog.csdn.net/qq_25241325/article/details/90753830这篇博文的 “四、检查是否安装成功”与“五、测试CUDA的Sample例子”

 

二 配置Cuda常见问题

1 第三步无法安装

提示"Ensure that there is enough space and the package is not corrupt" (类似的错误,具体记不清了)。

首先可能是包没有下载完全,首先进入路径查看包的大小对不对,如果比官网上写的大小差很多,请重新下载。比如我下载的Cuda 8.0官网上是1.4G,下载后的文件大小为1.5G。

其次可能是默认安装位置没有空余空间了,可以在安装过程中指定一个有空余的位置进行安装。我安装时在HOME新建了一个文件夹作为临时安装位置。

sudo sh cuda_8.0.61_375.26_linux.run --tmpdir=/home/davlee/temp

 2 ubuntu登录界面循环登录

现象是,卡在ubuntu的登录界面,输入正确的密码后电脑暂时黑屏,然后又回到登录界面,输入错误的密码就会提示密码错误。按Ctrl+Alt+f1可以进入tty1登录界面,输入用户名(本机为davlee)和密码后可以进入命令行。(可以再按Alt+f7回到图形界面)

问题可能出在安装CUDA时也安装了驱动,因此大致流程为:卸载驱动,关闭图形界面,重新安装CUDA,开启图形界面。具体流程请见https://blog.csdn.net/lihe4151021/article/details/90237681,或者自己搜索,这个问题很常见。

3 其他问题
(此处列举一些,我没遇到,但是可能会需要的解决方法)

1)禁用nouveau驱动:https://blog.csdn.net/qlulibin/article/details/78714596 

很多地方的步骤都会禁用这个驱动,所以谨慎起见可以禁用一下。另外,很多教程也会关闭图形界面,而在tty1中进行Cuda的安装,本文没有使用这样的方式,若想采取这样的方式请参考上面这个链接。(我觉得可能是Cuda版本较高时会有这个需要?)

 

Reference:

驱动版本对应Cuda版本: https://blog.csdn.net/BigData_Mining/article/details/103698420

https://blog.csdn.net/qq_25241325/article/details/90753830

https://blog.csdn.net/lihe4151021/article/details/90237681

赞,非常详细的安装Cuda教程:https://blog.csdn.net/wanzhen4330/article/details/81699769

赞,非常详细的安装Cuda教程:https://blog.csdn.net/qlulibin/article/details/78714596

posted @ 2020-09-04 17:48  hey,dummy  阅读(994)  评论(0编辑  收藏  举报