pytorch cuda安装

下载cuda,大概2G,一路确认安装
pytorch cuda安装_第1张图片

nvcc -V #查看安装是否成功
nvidia-smi #查看版本

pytorch cuda安装_第2张图片
pytorch官网查看复制pytorch安装命令
pytorch cuda安装_第3张图片

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

测试安装是否成功

import torch
torch.cuda.is_available() #输出True则安装成功

在这里插入图片描述

你可能感兴趣的:(python,python,cuda)