ubutun18.04安装TensorFlow GPU 1.15
时间: 2024-10-29 22:00:28 浏览: 41
Ubuntu 18.04版本安装TensorFlow GPU 1.15的步骤如下:
1. **更新系统**:
```
sudo apt-get update && sudo apt-get upgrade
```
2. **安装必要的库**:
```bash
sudo apt-get install -y software-properties-common python-software-properties
sudo add-apt-repository ppa:graphics-drivers/ppa
```
这一步骤将添加GPU驱动的PPA。
3. **安装CUDA和cuDNN**:
```bash
sudo apt-10.1
wget https://developer.nvidia.com/compute/cuda/10.1/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804-10-1-local-gcc-7-5-5-0-390.x86_64.run
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/repos/nvidia_repo_cuda/gpgkey
sudo apt-get update
sudo apt-get install -y cudnn7 libcudnn7-dev
```
4. **安装Python依赖**:
```bash
sudo apt-get install -y python3-pip python3-dev build-essential checkinstall libopenblas-dev luarocks
```
5. **安装TensorFlow-GPU**:
```bash
pip3 install tensorflow-gpu==1.15.0
```
如果遇到版本冲突或其他问题,你可以尝试指定特定版本的pip。
6. **检查安装**:
最后,确认安装是否成功,可以运行:
```bash
python3 -c "import tensorflow as tf; print(tf.__version__)"
```
**相关问题--:**
1. TensorFlow 1.15支持哪些GPU型号?
2. 安装过程中如何处理pip版本冲突?
3. 如何在Ubuntu 18.04上卸载已经安装的TensorFlow-GPU?
阅读全文
相关推荐

















