ubuntu安装tensorflow-gpu 和pytorch

本文参考了：

https://blog.csdn.net/qq_27825451/article/details/89082978

谢谢这位博主

这里我们选择清华源

sudo python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.2.0

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl

在选择tensorflow版本的时候，需要注意cuda cudnn的版本

下面是他们对应的版本：

相应网址：

https://www.tensorflow.org/install/source#common_installation_problems

https://www.tensorflow.org/install/source_windows

安装keras

pip install keras==2.1.5

(1)

安装virtualenv

sudo pip3 install -U virtualenv

创建空间

virtualenv --system-site-packages -p python3 ./tf2_py3

如果报错:

virtualenv --system-site-packages -p python3 ./tf2_py3 --no-setuptools

激活环境

source tf2_py3/bin/activate

安装tensorflow

pip install --upgrade tensorflow-gpu==2.0.0-alpha0

升级

sudo pip install --upgrade tensorflow-gpu--2.0.0-alpha0

测试

tf.test.is_gpu_available()

pytorch安装指导：

https://www.jianshu.com/p/5f8ec832ebb3

ubuntu安装tensorflow-gpu 和pytorch相关推荐

Ubuntu 20.04 安装Tensorflow GPU版本（GTX-1060显卡）
Ubuntu 20.04 安装Tensorflow GPU版本 (GTX-1060显卡) 参考: https://cyfeng.science/2020/05/02/ubuntu-install-nv ...
linux安装tensorflow教程,Ubuntu 16.04 安装 TensorFlow(GPU支持)
本文记录Ubuntu 16.04安装Tensorflow步骤,也包括怎么从源码编译安装Tensorflow. 要想安装Tensorflow GPU版本,你需要有一个新一点的Nvidia显卡. Tens ...
通过Anaconda在Ubuntu16.04上安装 TensorFlow(GPU版本)
一. 安装环境 Ubuntu16.04.3 LST GPU: GeForce GTX1070 Python: 3.5 CUDA Toolkit 8.0 GA1 (Sept 2016) cuDNN v6 ...
Ubuntu安装tensorflow报错：tensorflow-xx.whl not a supported wheel on this platform
解决Ubuntu安装tensorflow报错:tensorflow-0.5.0-cp27-none-linux_x86_64.whl is not a supported wheel on this ...
windows10下安装tensorflow(gpu版本)
windows10安装tensorflow的gpu版本(pip3安装方式) 前言: TensorFlow 有cpu和 gpu两个版本:gpu版本需要英伟达CUDA 和 cuDNN 的支持,cpu版本不 ...
Docker 安装 TensorFlow GPU 实战
2019独角兽企业重金招聘Python工程师标准>>> Docker 安装 TensorFlow GPU 实战博客分类: 深度学习 docker 安装背景 AI如雨后春笋般的出现, ...
GTX860M安装Tensorflow gpu踩坑
GTX860M安装Tensorflow gpu踩坑前几天给台式成功配置了lightgbm的gpu版本和tensorflow的gpu版本,但是在单位还是要用LSTM,写一篇对应版本的踩坑文章日后查阅. ...
ubuntu 16.04 安装TensorFlow GPU版本
推荐新版安装教程 http://blog.csdn.net/chenhaifeng2016/article/details/78874883 在ubuntu 16.04上安装cuda8.0和cudnn ...
ubuntu16.04下安装TensorFlow(GPU加速)----详细图文教程【转】
本文转载自:https://blog.csdn.net/zhaoyu106/article/details/52793183 le/details/52793183 写在前面一些废话接触深度学习已 ...
Deepin 安装tensorflow gpu亲测教程（适用于双显卡）
目录前言 Let's GO Step 1: 安装Python Step 2: 切换到大黄蜂显卡驱动解决方案 Step 3: 安装源内版本的CUDA Step 4: 安装cuDNN和NCCL Step ...

ubuntu安装tensorflow-gpu 和pytorch

ubuntu安装tensorflow-gpu 和pytorch相关推荐

最新文章

热门文章