Pytorch-GPU

参考文章：Pytorch-gpu安装（CUDA11.1 + MX450 + Win10）-CSDN博客

一、电脑环境

通过“NVIDIA控制面板”查看CUDA版本，此处的CUDA版本意为最高可支持的版本，例如我的最高支持到12.3

image-20240118123121876

二、安装CUDA

目前Pytorch的支持最高版本是12.1，因此我CUDA虽然最高支持到12.3，但也需下载12.1，否则后期跑代码会有报错

验证：

1
2
3
4
5
6
❯ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Wed_Feb__8_05:53:42_Coordinated_Universal_Time_2023
Cuda compilation tools, release 12.1, V12.1.66
Build cuda_12.1.r12.1/compiler.32415258_0

下载链接：CUDA Toolkit 12.1 Downloads | NVIDIA Developer

三、安装对应版本cudnn（可选非必需）

下载12.x版本的cudnn，是个压缩包，里面有bin、include、lib三个文件夹，复制到CUDA对应的安装目录下

下载链接：cuDNN Download | NVIDIA Developer

四、安装Pytorch

image-20240118144006183

以CUDA 12.1版本为例，Conda包管理：

1
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

pip包管理：

1
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

下载链接：PyTorch

五、验证

在虚拟环境中，进入python交互环境，如果显示True即为成功

image-20240118142338107