换了系统之后需要重新配备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就表示安装成功了。