Pytorch

v1.7.1

安装

conda install pytorch torchvision -c pytorch

使用

import torch
# 检查GPU是否支持CUDA
cudaAvailable = torch.cuda.is_available()
print(cudaAvailable)