tensorflow1.x gpu安装清华源
时间: 2024-01-14 14:04:04 浏览: 92
以下是在 Ubuntu 系统下使用清华源安装 TensorFlow 1.x GPU 版本的步骤:
1. 添加清华源
打开终端,输入以下命令:
```
sudo cp /etc/apt/sources.list /etc/apt/sources.list.bak # 备份原 sources.list 文件
sudo gedit /etc/apt/sources.list
```
在文本编辑器中将原内容全部替换为以下内容:
```
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-updates main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-backports main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-security main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ xenial main
deb https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ xenial main
```
保存并退出。
2. 安装依赖
在终端中输入以下命令:
```
sudo apt-get update
sudo apt-get install python3-pip python3-dev python-virtualenv
sudo apt-get install libcupti-dev
```
3. 创建虚拟环境
在终端中输入以下命令:
```
virtualenv --system-site-packages -p python3 ~/tensorflow-gpu
```
其中,`~/tensorflow-gpu` 为虚拟环境的名称,可以自行修改。
4. 激活虚拟环境
在终端中输入以下命令:
```
source ~/tensorflow-gpu/bin/activate
```
5. 安装 TensorFlow
在终端中输入以下命令:
```
pip3 install --upgrade tensorflow-gpu==1.15
```
等待安装完成即可。
6. 验证安装
在终端中输入以下命令:
```
python
```
进入 Python 解释器后输入以下代码:
```
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
```
如果没有报错,则说明安装成功。
注意:如果在使用 TensorFlow 时遇到 CUDA 相关的问题,可以参考以下链接进行解决:
https://blog.csdn.net/weixin_41890118/article/details/82757829
阅读全文