如何在Ubuntu上安装tensorflow和pytorch

1. 安装pytorch

1. pycharm中选择新建项目,选择在conda配置环境
2. 新建一个main.py
3. 在终端上输入代码：conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
4. 过程中输入yes即可安装

验证是否安装成功

import torch 
print(torch.__version__)
print(torch.cuda.is_available()) 
print(torch.cuda.device_count()) 
print(torch.backends.cudnn.version()) 
print(torch.version.cuda) 
quit()

2. 安装tensorflow

1. pycharm中选择新建项目，选择在conda配置环境
2. 新建一个main.py
3. 在终端上输入代码：

conda install --channel https://mirrors.ustc.edu.cn/anaconda/pkgs/main/ tensorflow-gpu==2.1.0

pip install tensorflow-gpu==2.1.0 -i https://pypi.mirrors.ustc.edu.cn/simple/

pip install tensorflow-gpu==2.1.0 -i http://pypi.douban.com/simple/

pip install tensorflow-gpu==2.1.0 -i http://mirrors.aliyun.com/pypi/simple/

4. 过程中输入yes即可安装

如果是2.1.0，python必须是3.7

验证是否安装成功

import tensorflow as tf 
tf.test.is_built_with_cuda()
quit()

注意

镜像源可以多换几个，之前清华的一直下不下来
清华：https://pypi.tuna.tsinghua.edu.cn/simple
阿里云：http://mirrors.aliyun.com/pypi/simple/
中国科技大学： https://pypi.mirrors.ustc.edu.cn/simple/
豆瓣：http://pypi.douban.com/simple/

出现一个降级protobuf

解决方式：

pip uninstall protobuf

pip install protobuf==3.19.0