如何正确下载PyTorch、CUDA版本！！

因为在torch框架里经常出现NVIDIA、CUDA、PyTorch这几个不兼容，等等一些让人头疼的问题。这里总结正确下载pytorch的方法。

安装pytorch

一.查看自己cuda版本

方法一：

方法二：

CMD中输入：

nvidia-smi

我的是CUDA版本12.0，版本向下兼容，意思就是CUDA 12.0及以下版本的都可以安装

二.打开pytorch的官网：PyTorchAn open source machine learning framework that accelerates the path from research prototyping to production deployment.https://pytorch.org/（官网网址：https://pytorch.org/）

复制代码：

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

安装CUDA

官网：CUDA Toolkit 11.7 Downloads | NVIDIA DeveloperResources CUDA Documentation/Release NotesMacOS Tools Training Sample Code Forums Archive of Previous CUDA Releases FAQ Open Source PackagesSubmit a BugTarball and Zip Archive Deliverableshttps://developer.nvidia.com/cuda-11-7-0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local

选择 11.7的对应的进入安装，有点大2.5g，耐心等待呀

打开exe，自定义

全选

记住安装位置，默认位置

配置环境变量

在系统变量里新建这四个，后边是v11.7安装的位置

CUDA_PATH
CUDA_PATH_V11_6
NVCUDASAMPLES_ROOT
NVCUDASAMPLES11_0_ROOT

验证使用CMD查看CUDA是否安装成功命令：nvcc --version或nvcc -V

nvcc --version
nvcc -V

安装成功！！