3090 搭建conda虚拟环境TensorFlow+pytorch(自用)

在使用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())

你可能感兴趣的:(环境配置,tensorflow,pytorch,深度学习)