ubuntu16.04配置cuda8.0, 9.0切换

之前用pytorch,所以已经配置好了cuda9.0和cudnn7.0,但是在pip3 install tensorflow-gpu安装成功后,测试tensorflow的例子是报错了,两个错误先后报出:
ImportError:libcudart.so.8.0: cannot open shared object file: No such fileor directory
ImportError:libcudnn.6.0: cannot open shared object file: No such file ordirectory

查了一下原因是因为tensorflow-gpu1.4.1的版本默认支持cuda8.0和cudnn6.0,也不知道如何修改tensorflow的配置,所以只能再安装cuda8.0和cudnn6.0了

安装cuda8.0(按照官方的安装说明即可)

注意这里下载的是cuda8.0的runfile(local)文件。下载完cuda8.0后,执行如下语句,运行runfile文件:
sudo sh cuda_8.0.27_linux.run

因为驱动之前已经安装,这里就不要选择安装驱动。其余的都直接默认或者选择是即可。 使用:
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

安装cudnn6.0

cudnn安装步骤:
从官网上下载cudnn的安装包,将安装包解压,将此安装包放在home路径下即可,并在当前路径下进行解压,解压后的文件夹名为cuda。
在终端上编辑如下代码:
(也就是把include和lib64的文件都分别拷贝到/usr/local/cuda/include/和/usr/local/cuda/lib64/中)
cd cuda/include
sudo cp cudnn.h /usr/local/cuda/include/
cd cuda/lib64
sudo cp libcudnn* /usr/local/cuda/lib64/
再运行tensorflow的例子,就成功了。

你可能感兴趣的:(ubuntu16.04配置cuda8.0, 9.0切换)