在使用3090显卡的服务器上,搭建TensorFlow和pytorch环境
首先,在anaconda创建虚拟环境,本次实验的Python语言版本为3.8
conda create -n sum python=3.8
进行安装TensorFlow操作
conda install tensorflow-gpu
conda list显示如下
进行安装pytorch操作,一般安装pytorch需要访问pytorch官网,这里直接放出适合3090显卡物理驱动的安装语言
pip install torch===1.7.1+cu110 torchvision===0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
conda list显示如下
分别测试能否使用gpu
import tensorflow as tf
tf.config.list_physical_devices('GPU')
import torch
print(torch.cuda.is_available())