一、检查虚拟环境中Pytorch的版本
1.进入虚拟环境
2.输入python
3.输入
import torch
print(torch.__version__)
二、判断安装的pytorch是GPU版本还是CPU版本,指令如下:
True代表是GPU版本,False代表是CPU版本
三、查看可用的CUDA数量
import torch #如果pytorch安装成功即可导入
print(torch.cuda.device_count()) #查看可用的CUDA数量
四、查看CUDA版本的另外一种方式,在命令行中输入如下命令:
nvcc -V
nvcc --version
或者使用nvidia-smi,也可以查看CUDA的版本