本文參考了：

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的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。