conda版本
conda --version
GPU版本
//注意:中间没有空格
nvidia-smi
CUDA版本
//注意:中间有空格,第二个V大写
nvcc -V
pytorch版本
//第一步输入
python
//第二步输入
import torch
torch.__version__ //注意:下划线是两条!!!
checkPytorch1.png
checkPytorch2.png
验证GPU版pytorch是否可用
python
import torch
torch.cuda.is_available()
//返回True代表可用
torch.cuda.is_available().png
torchvision版本
//第一步输入
python
//第二步输入
import torchvision
torchvision.__version__ //注意:下划线是两条!!!
网友评论