conda版本
conda --version
GPU版本
//注意:中間沒有空格
nvidia-smi
CUDA版本
//注意:中間有空格,第二個V大寫
nvcc -V
pytorch版本
//第一步輸入
python
//第二步輸入
import torch
torch.__version__ //注意:下劃線是兩條E棵凇!嗜憔!
checkPytorch1.png
checkPytorch2.png
驗(yàn)證GPU版pytorch是否可用
python
import torch
torch.cuda.is_available()
//返回True代表可用
torch.cuda.is_available().png
torchvision版本
//第一步輸入
python
//第二步輸入
import torchvision
torchvision.__version__ //注意:下劃線是兩條<贰!呐舔!