LInux系统中深度学习之pytorch安装测试

一、在conda中新建虚拟环境

conda create -n pytorch python=3.7

二、安装

conda install pytorch torchvision cudatoolkit=10.1

pip install torch

三、测试

import torch
x = torch.rand(5, 3)
print(x)

print(torch.cuda.is_available())

print(torch.cuda.device_count())

print(torch.cuda.get_device_name(0))

print(torch.cuda.current_device())