PyTorch GPU版本安装方法,请看anaconda虚拟环境下镜像安装PyTorch,利用jupyter notebook访问虚拟环境
检测程序如下:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
运行结果:
1.4.0
True
PyTorch版本为1.4.0
True代表GPU版本成功安装!
PyTorch GPU版本安装方法,请看anaconda虚拟环境下镜像安装PyTorch,利用jupyter notebook访问虚拟环境
检测程序如下:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
运行结果:
1.4.0
True
PyTorch版本为1.4.0
True代表GPU版本成功安装!