PyTorch GPU版/CUDA 环境详细安装教程(完美避坑)

一、安装 CUDA 

1、安装CUDA前请先确认Torch 所支持的版本，以免安装的Torch 不支持最新CUDA的版本，下图可见Torch所支持的CUDA最新版本位 11.7，如果我们安装CUDA 12.0 版本 执行 `torch.cuda.is_available()` 时则为False。

PyTorch版本查询链接： Start Locally | PyTorch

 

 

 

2、下载安装

根据上图所示，选择合适的CUDA版本，安装CUDA  C盘需要5 GB左右的空间，如果C盘空间少可使用分区助手来扩容，详细的安装过程不在概述，傻瓜试一键安装。

CUDA下载链接：CUDA Toolkit Archive | NVIDIA Developer

 二、安装PyTorch

 1、复制 `Run this Command` 这里的命令，在CMD中执行即可