ubuntu18.04配置python中的tensorflow-gpu环境

换了系统之后需要重新配备python的环境，我的电脑系统是ubuntu18.04，GPU10.1版本，打算装3.6版本的python。

搜索tensorflow版本

conda search tensorflow

查看安装源

conda config --show channels

添加清华源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

安装tensorflow

 conda install tensorflow-gpu==1.14

相应的cuda和cudnn版本是： cudatoolkit=10.1.243 , cudnn=7.6.5 ，在安装tensorflow时，这两个库会默认被安装，需要注意看更新的软件包，尤其是python的版本。

在电脑软件与更新上修改附加驱动驱动版本到435版本

安装pytorch

进入官网https://pytorch.org/
选择安装命令并加上清华源：

pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f 
https://download.pytorch.org/whl/torch_stable.html 
 -i  https://pypi.tuna.tsinghua.edu.cn/simple some-package

大概有7百多兆，网好的话一会就下完了, 通过输入以下语句进行测试

import torch
torch.cuda.is_available()

结果是True就表示安装成功了。

end

参考来源

• https://blog.csdn.net/gdymind/article/details/82153643