检查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