Linux非root用戶如何優雅的安裝cuda和cudnn

發布時間：2018-05-23 16:11,

瀏覽次數：804

, 標簽：

Linux

root

cuda

cudnn

Linux非root用戶如何優雅的安裝cuda和cudnn

跑深度學習模型時，總會遇到各種版本的問題。如深度學習框架與cuda的版本不符，或者管理安裝的cuda的版本太低，因此作為非root用戶就很難受。為自己裝cuda和深度學習框架就非常有必要

* anaconda的必要性

* 非root安裝cuda和cudnn

anaconda

* python有一個很好的版本管理器——anaconda，安裝它就可以自行管理各種python版本

* anaconda 安裝和使用

非root用戶安裝cuda和cudnn

cuda的root用戶安裝，網上的教程非常多，但是非root用戶安裝的教程就很難找到

1、cuda下載 https://developer.nvidia.com/cuda-downloads

.

2、在選擇linux及對應的系統之后，選擇 runfile(local)這個選項，然后下面給出的命令 “sudo sh

cuda_9.0.176_384.81_linux.run” 中sudo這個需要管理員賬號，其實可以去掉sudo，直接運行，然后按照給出的提示同意協議

3、linux沒有圖形界面時，可以在瀏覽器復制下載鏈接“link”，然后再linux命令 “wget link”,下載下來filename.run

4、chmod +x filename.run 然后 ./filename.run

5、在協議中選擇同意EULA(accept),不安裝driver installation

(no),然后再安裝cuda時選擇個人用戶的目錄，如/home/yourname/cuda9,以及cudasamples的目錄

6、cudnn的安裝,從官網下載https://developer.nvidia.com/cudnn

(需要注冊賬號),

我下載的是cudnn-9.0-linux-x64-v7.tgz，然后用tar -xzvf cudnn-9.0-linux-x64-v7.tgz

解壓，解壓到了cuda文件夾。(cd cuda && make)可選命令

7、拷貝過去cudnn->cuda (cuda9是個人用戶的下的目錄/home/yourname/cuda9)

cp cuda/include/cudnn.h cuda9/include/

cp cuda /lib64/libcudnn* cuda9/lib64

chmod a+r cuda9/include/cudnn/h cuda9/lib64/libcudnn*

8、修改個人用戶的環境變量

環境變量文件 ~/.bashrc位于home/yourname/~/.bashrc(用vi ~/.bashrc編輯)

“export PATH=$HOME/cuda9/bin:$PATH”

“export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$HOME/cuda9/lib64/”

這兩條命令添加進去HOME目錄就是/home/yourname

修改之后 source ~/.bashrc使環境變量生效

快速測試

import tensorflow as tf x = tf.nn.conv2d(tf.ones([1,1,10,1]),

tf.ones([1,5,1,1]), strides=[1, 1, 1, 1], padding='SAME') with tf.Session() as

sess: sess.run(x) # this should output a tensor of shape (1,1,10,1) with

[3,4,5,5,5,5,5,5,4,3]

總結

以上是生活随笔為你收集整理的不安装cudnn可不可以_Linux非root用户如何优雅的安装cuda和cudnn的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。