Pytorch安装-编程知识

Pytorch安装

一、查看 CUDA 版本
使用pip install torch (2.0.0版本)，这样安装的torch是直接运行在CPU上的，想要使用GPU版本需要使用对应的cuda版本。
在安装pytorch时我们需要选择对应CUDA版本的pytorch，那如何查看CUDA版本呢？

1.NVIDIA方式
打开NVIDIA控制面板
在这里插入图片描述
点击系统信息—>点击组件

在这里插入图片描述 2.cmd 方式

nvidia-smi

在这里插入图片描述
二、去Pytorch官网下载对应CUDA版本的pytorch
Pytorch官网：点这里

尽管pytorch官网上目前只给出了11.8的Cuda支持，但是社区明确表明了兼容高版本Cuda。在这里插入图片描述安装命令：

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

首先激活Python 虚拟环境
conda activate test
在这里插入图片描述
输入安装命令：

在这里插入图片描述等待完成，有点慢，2.6G左右…

安装好后，检验是否安装成功，新建.py文件

import torch
print(torch.__version__)
print(torch.version.cuda)#cuda版本
print(torch.backends.cudnn.version())
print(torch.cuda.is_available()) #cuda是否可用，返回为True表示可用
print(torch.cuda.device_count())#返回GPU的数量
print(torch.cuda.get_device_name(0))#返回gpu名字，设备索引默认从0开始