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