[Python知识库]anaconda安装pytorchgpu版

先看一下cuda版本（桌面点击右键，打开nvidia控制面板，点详细信息，点组件）

打开pytorch官网

?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())