Pytorch GPU深度学习环境安装

Pytorch GPU深度学习环境安装

PS：Python版本3.8

查看自己电脑中是否有NVIDIA显卡

右键此电脑点击属性

点击设备管理器

找到显示适配器

观察显示适配器中是否有NVIDIA显卡，如果有则说明有独立显卡，否则说明无独立显卡。

检查电脑中是否安装显卡驱动

打开控制台输入nvidia-smi，并查看可用的CUDA最高版本。

找到合适的Pytorch版本

打开官网地址，找到适合的Pytorch安装版本，要求CUDA版本低于本机显卡支持的最高CUDA版本。

因为我本机的电脑能支持的CUDA最高版本为11.4，所以我只能安装低于11.4版本的Pytorch，如图片中的10.2、11.3均可，但是11.6不能安装。还有一点要注意的是，如果你电脑中无GPU，则只能CPU版本的Pytorch。

复制相关下载代码进行下载即可

例如我这里安装CUDA 11.3的PytorchV1.12.0，应该使用以下语句即可。

conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c pytorch

验证是否安装成功

打开解释器输入以下代码：

import torch

torch.cuda.is_available()

返回True说明安装成功，并且深度学习环境可用。