这个屌丝很懒，什么也没留下！

【pytorch】cuda torch torchvision版本对应_torch1.13.1对应cuda

作者：AllinToyou | 2024-03-05 21:25:19

torch1.13.1对应cuda

在这里插入图片描述
查看服务器cuda版本：

cat /usr/local/cuda/version.txt
1

查看torch torchvision版本：

import torch
print(torch.__version__)

import torchvision
print(torchvision.__version__)

print(torch.cuda.is_available()) # 查看是否有可用GPU
print(torch.cuda.device_count()) # 查看GPU数量
1
2
3
4
5
6
7
8

本文内容由网友自发贡献，转载请注明出处：【wpsshop博客】

【pytorch】cuda torch torchvision版本对应_torch1.13.1对应cuda

cuda torch torchvision版本对应以及查看