防坑 centos7 安装 CUDA9.0 + cudnn7.1 +TensorFlow GPU版1.6.0/1.8.0

4 篇文章 0 订阅
2 篇文章 0 订阅

为了加快模型的训练和提高效率,于是乎入手TensorFlow的GPU版!

一、安装CUDA

下载地址:https://developer.nvidia.com/cuda-90-download-archive

首先检查自己的电脑有没有NVIDIA的显卡

检测环境:传输门 CUDA官方安装说明https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html

强烈建议大家按照官方的安装说明一步步走,先检测环境,在下载相应的版本,一定可以安装好cuda。

这里有一个坑就是,如果你是双显卡,只是用nvida来跑程序,那么在安装cuda的时候

1、sudo sh xxxxx.run(cuda的安装程序,选择runfile)
2、accpet
3、接下来是一堆无关重要的说明文档,按q直接跳过,如果你要细究估计要花1h以上读这个文档!!!
4、注意了,接下来就选择安装的部分,第一个是nvidia的驱动,如果已经安装就n,如果没有就y。

5、现在坑来了,如果你想回到图形化桌面来,这个openGL就选择n,不要安装!!!!

6、接下就是一些例子,自己选择!然后安装ok

二、下载cudnn

下载地址:https://developer.nvidia.com/rdp/cudnn-archive

去官网下载对应版本的cudnn

安装官方的使用说明https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html就可以了!!!!

三、安装TensorFlow

推荐使用pip安装

这里也有一个坑那就是tensorflow的版本一定要与cuda的版本对应,不然白搭!所以在安装cuda前要,选好版本!

简单来说:tf1.6以上用只能是cuda9.0,其他的tf版本就是cuda8.0等,最好自己去查查!可恶的是tf官方和nVidia都没有版本对应的说明!!!

不过,我最后发现cuda的不同版本是可以并存的!只要做好软件链就ok!!!

祝大家好运!!

  • 2
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
您好!要在CentOS 7上安装CUDA 10.2和cuDNN 8,您可以按照以下步骤进行操作: 1. 检查您的显卡是否支持CUDA 10.2,并确保您已经安装了适当的驱动程序。 2. 前往NVIDIA官网下载CUDA Toolkit 10.2的安装包(https://developer.nvidia.com/cuda-10.2-download-archive)。选择合适的本(rpm或runfile)并下载到您的系统上。 3. 安装CUDA Toolkit 10.2。如果您下载的是rpm文件,请使用以下命令安装: ``` $ sudo rpm -i cuda-repo-<version>.rpm $ sudo yum clean all $ sudo yum install cuda ``` 如果您下载的是runfile文件,请使用以下命令安装: ``` $ sudo sh cuda_<version>_linux.run ``` 根据提示进行安装并接受许可协议。 4. 等待CUDA Toolkit的安装完成后,添加CUDA的环境变量。打开终端并编辑`~/.bashrc`文件,添加以下行: ``` export PATH=/usr/local/cuda-10.2/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH ``` 保存文件后,运行`source ~/.bashrc`使其生效。 5. 下载cuDNN 8,您需要先注册NVIDIA开发者计划(如果尚未注册),然后前往https://developer.nvidia.com/rdp/cudnn-archive下载cuDNN 8对应的本。 6. 将下载cuDNN 8压缩文件解压缩,并将文件复制到CUDA安装目录中。打开终端,运行以下命令: ``` $ tar -xzvf <path_to_cudnn_tar> $ sudo cp cuda/include/cudnn*.h /usr/local/cuda-10.2/include $ sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.2/lib64 $ sudo chmod a+r /usr/local/cuda-10.2/include/cudnn*.h /usr/local/cuda-10.2/lib64/libcudnn*

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值