目录
目录
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