先看一下cuda版本(桌面点击右键,打开nvidia控制面板,点详细信息,点组件)
打开pytorch官网
打开anaconda prompt
conda create -n mytorch python=3.6
创建名为mytorch,python版本3.6的环境
conda activate mytorch
激活环境
将刚刚pytorch官网得到的安装代码复制上,即上图蓝色部分,安装即可
安装完测试一下
在mytorch环境里输入python
再将下面代码复制上
import torch
flag = torch.cuda.is_available()
print(flag)
ngpu= 1
# Decide which device we want to run on
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3,3).cuda())