ubuntu安装tensorflow-gpu和pytorch-gpu

安装好anaconda后(我装的anaconda的base的python版本是3.8的，但没影响)

安装tensorflow-gpu

conda create -n tensorflow python=3.7

conda activate tensorflow

conda install tensorflow-gpu

就好了

这里默认的是安装tensorflow2.0

安装tensorflow1.0

conda create -n tensorflow1.0 python=3.6

conda activate tensorflow1.0

pip install tensorflow-gpu==1.13.1 -i https://pypi.tuna.tsinghua.edu.cn/simple

这里要注意tensorflow的版本要与安装的cuda10.0对应

最开始我安装了tensorflow-gpu=1.4.0和1.12.0都不行，安装完后倒入报错，一查发现它们只支持cuda9.0

注意安装tensorflow1.0的话要pyhon3.6或3.5，3.7不行

安装pytorch-gpu

conda create -n pytorch python=3.7

conda activate pytorch

pip install torch==1.6.0 torchvision==0.7.0 -f http://download.pytorch.org/whl/torch_stable.html

下的比较慢，当时是当它下了一晚上

安装完之后就好了