0
点赞
收藏
分享

微信扫一扫

ubuntu18.04安装cuda10.0+cudnn+tensorflow1.x

芭芭蘑菇 2022-04-01 阅读 128
ubuntu

目录

目录

1.cuda安装

测试

2.cudnn安装

debug

卸载cudnn

卸载cuda

3.安装tensorflow

测试

debug


1.cuda安装

CUDA Toolkit 10.0 Archive | NVIDIA Developer

下载好runfile文件后在目录下输入

sudo sh cuda_10.0.130_410.48_linux.run

空格一直到最下 不要选驱动!!!

加入.bashrc环境

sudo gedit .bashrc

export PATH="/usr/local/cuda-10.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/lcoal/cuda-10.0/lib64:$LD_LIBRARY_PATH"

测试

cd /usr/local/cuda/samples/1_Utilities/deviceQuery 
sudo make
./deviceQuery

2.cudnn安装

cuDNN Archive | NVIDIA Developer

下载除了第二个

目录下解压
tar -zxvf cudnn-10.0-linux-x64-v7.3.1.20.tgz 
 
链接
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
 
安装
sudo dpkg -i libcudnn7_7.3.1.20-1+cuda10.0_amd64.deb 
sudo dpkg -i libcudnn7-dev_7.3.1.20-1+cuda10.0_amd64.deb 
sudo dpkg -i libcudnn7-doc_7.3.1.20-1+cuda10.0_amd64.deb 

测试
cp -r /usr/src/cudnn_samples_v7/ ~ && cd ~/cudnn_samples_v7/mnistCUDNN
make clean && make && ./mnistCUDNN

debug

若输入./mnistCUDNN后报错:
./mnistCUDNN: error while loading shared libraries: libcudart.so.9.0: cannot open shared object file: No such file or directory

sudo cp /usr/local/cuda-10.0/lib64/libcudart.so.10.0 /usr/local/lib/libcudart.so.10.0 && sudo ldconfig
sudo cp /usr/local/cuda-10.0/lib64/libcublas.so.10.0 /usr/local/lib/libcublas.so.10.0 && sudo ldconfig
sudo cp /usr/local/cuda-10.0/lib64/libcurand.so.10.0 /usr/local/lib/libcurand.so.10.0 && sudo ldconfig

卸载cudnn

sudo rm -rf /usr/local/cuda/include/cudnn.h 
sudo rm -rf /usr/local/cuda/lib64/libcudnn*

卸载cuda

sudo /usr/local/cuda-10.0/bin/cuda-uninstaller
sudo rm -rf /usr/local/cuda-10.0

3.安装tensorflow

创建虚拟环境

sudo apt-get install python3-pip python3-dev python-virtualenv
virtualenv --system-site-packages ~/***

进入

source ~/***/bin/activate

安装tensorflow==1.14.0

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.14.0

测试

import tensorflow as tf
hello = tf.constant("Hello TensorFlow-GPU!!")
sess = tf.Session()
print(sess.run(hello))

debug

如果出现这个错误则更改版本

    class DescriptorBase(metaclass=DescriptorMetaclass):
                                  ^
SyntaxError: invalid syntax
pip uninstall protobuf==3.18.0
pip install protobuf==3.17.3
举报

相关推荐

0 条评论