ubuntu16.04+CUDA本地安装+cuDNN本地安装+tensorflow

最新推荐文章于 2024-04-15 17:24:46 发布

原创最新推荐文章于 2024-04-15 17:24:46 发布 · 1.3k 阅读

CC 4.0 BY-SA版权

文章标签：

37 篇文章

订阅专栏

本文详细介绍了如何在Ubuntu系统上安装CUDA 8.0，包括显卡驱动的安装、环境变量配置、cuDNN集成及TensorFlow-GPU版本的安装测试。适合初学者快速上手。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

GPU card with CUDA Compute Capability 3.0 or higher.

1、下载cuda_8.0.61_375.26_linux.run后放入soft文件夹，建立local文件夹（cuda安装路径）

2、运行“sh cuda_8.0.61_375.26_linux.run”，

3、在ubuntu系统中已经有显卡驱动时（利用nvidia-smi命令检测），跳过第一步显卡驱动安装，直接到cuda tools安装。为了省时间，然后跳过cuda sample安装。

driver online install: sudo apt-get install nvidia-384 or nvidia-375

OR:

下载完NVIDIA.run后，执行如下语句，运行runfile文件：

sudo sh NVIDIA.run

4、该命令结束后，添加该安装相关路径到用户环境内：

打开./.bashrc文件，下面加上

export PATH="local/cuda-8.0/bin:$PATH"

export LD_LIBRARY_PATH="local/cuda-8.0/lib64:$LD_LIBRARY_PATH"

然后运行source ~/.bashrc，使之生效

运行nvcc -V可以测试安装是否成功

注意：

如果安装系统环境下需要添加local/cuda-8.0/lib64到/etc/ld.so.conf，然后运行ldconfig使之生效。

sudo gedit /etc/profile

打开“profile”文件，在末尾处添加（注意不要有空格，不然会报错):

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

重启电脑：

sudo reboot

至此cuda本地安装成功。

5、本地安装cudnn

实际cudnn是特别用于神经网络的加速库，依然依托于cuda tools平台。安装过程就是复制cudnn头文件到cuda的include路径，复制库文件(.so)到cuda的lib64路径。

cp cuDNN/cuda/include/cudnn.h local/cuda/include/

cp cuDNN/cuda/lib64/* /usr/local/cuda/lib64/

cd local/cuda/lib64/

可能需要删除原cudnn库
sudo rm -rf libcudnn.so libcudnn.so.X.X（都是软连接）

然后修改文件权限，并创建新的软连接

由于我所用tensorflow-gpu依赖cuDNN6.0,所以安装的cuDNN6.0

更改库文件权限：

chmod u=rwx,g=rx,o=rx libcudnn.so.6.0.20 
建立软连接

ln -s libcudnn.so.6.0.20 libcudnn.so.6.0
ln -s libcudnn.so.6.0 libcudnn.so

6、tensorflow-gpu在线安装

$ pip install tensorflow-gpu  # Python 2.7;  GPU support
$ pip3 install tensorflow-gpu # Python 3.n; GPU support

7、测试tensorflow-gpu

http://www.linuxidc.com/Linux/2015-04/116445.htm

http://www.linuxidc.com/Linux/2016-12/138906.htm

http://blog.youkuaiyun.com/masa_fish/article/details/51882183

http://blog.youkuaiyun.com/t5131828/article/details/53258925

http://blog.youkuaiyun.com/a1154761720/article/details/52302595

http://www.linuxidc.com/Linux/2016-12/138906.htm

http://blog.youkuaiyun.com/u013645510/article/details/53448806