查看PyTorch的版本及CUDA和cuDNN版本
检查PyTorch版本torch.version# PyTorch versiontorch.version.cuda# Corresponding CUDAversiontorch.backends.cudnn.version()# CorrespondingcuDNN versiontorch.cuda.get...
·
检查PyTorch版本
-
torch.version # PyTorch version
-
torch.version.cuda # Corresponding CUDA version
-
torch.backends.cudnn.version() # Corresponding cuDNN version
-
torch.cuda.get_device_name(0) # GPU type
更新PyTorch
- conda update pytorch torchvision -c pytorch
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
7
0- 0
已为社区贡献5条内容
所有评论(0)