ubuntu18.04安装cuda8+cudnn+tensorflow
时间: 2023-12-09 14:02:34 浏览: 192
ubuntu18.04+安装GPU++CUDA+cuDNN
好的,下面是安装步骤:
1. 安装CUDA 8.0
首先,你需要下载CUDA 8.0安装包,可以在官网上下载或者使用以下命令:
```
wget https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda_8.0.61_375.26_linux-run
```
下载完成后,运行以下命令进行安装:
```
sudo sh cuda_8.0.61_375.26_linux-run
```
安装过程中需要你进行一些选择,可以按照默认选项进行。
安装完成后,在`~/.bashrc`文件中添加以下环境变量:
```
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
```
然后运行以下命令使配置生效:
```
source ~/.bashrc
```
2. 安装cuDNN
cuDNN是NVIDIA提供的深度学习库,可以加速深度学习的训练和推理过程。你需要先在官网上注册一个账号,然后才能下载cuDNN。
下载完成后,解压文件并将文件复制到CUDA的安装路径下:
```
tar -xzvf cudnn-8.0-linux-x64-v6.0.tgz
sudo cp -P cuda/include/cudnn.h /usr/local/cuda-8.0/include
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64/
sudo chmod a+r /usr/local/cuda-8.0/include/cudnn.h /usr/local/cuda-8.0/lib64/libcudnn*
```
3. 安装TensorFlow
最后,你可以使用pip安装TensorFlow:
```
pip install tensorflow-gpu==1.3
```
如果你想使用CPU版本的TensorFlow,可以使用以下命令:
```
pip install tensorflow==1.3
```
安装完成后,你可以测试一下TensorFlow是否正常工作:
```
python
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
```
如果输出了`Hello, TensorFlow!`,那么TensorFlow就安装成功了。
阅读全文