在PyTorch中使用CUDA，根据你的具体环境和需求调整版本号，确保安装的PyTorch版本与你的CUDA版本兼容。

在PyTorch中使用CUDA，你需要确保正确安装了匹配你的GPU的CUDA Toolkit。以下是在PyTorch中使用CUDA的一般步骤：

1. 检查CUDA支持： 首先，确保你的GPU支持CUDA。你可以在官方CUDA支持列表上查找你的GPU型号。或者直接命令行
2. 安装CUDA Toolkit： 下载并安装与你的GPU型号匹配的CUDA Toolkit。你可以从NVIDIA官网下载。在安装期间，可以选择安装适用于你的系统的CUDNN库。
3. 安装cuDNN（可选）： cuDNN是NVIDIA的深度神经网络库，可以加速深度学习任务。在CUDNN下载页面下载适用于你的CUDA版本的cuDNN，并按照安装说明进行安装。
4. 安装PyTorch： 选择合适的PyTorch版本并使用pip或conda进行安装。按下面步骤2执行。

如果电脑已经安装过CUDA Toolkit和cuDNN,则步骤如下：

1 查看本机CUDA版本

输入命令

NVIDIA-SMI

如下，CUDA版本11.6

2 查看对应CUDA的对应pytorch版本安装

官网查看对应系统对应cuda版本对应pytorch版本的安装命令，链接如下

Previous PyTorch Versions | PyTorchAn open source machine learning framework that accelerates the path from research prototyping to production deployment.https://pytorch.org/get-started/previous-versions/如图：

（一般pip安装会比conda安装较高效）。

3 用pip 安装

本文安装 torch 1.13.0+cuda11.6 ，命令如下

pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116

注意：

这个命令将安装PyTorch、torchvision和torchaudio，并指定CUDA版本为11.6（注意：PyTorch的版本和CUDA版本可能会更新，确保使用适用于你系统的版本）。

请注意，使用pip安装时，你需要确保你已经安装了正确版本的CUDA Toolkit。如果你的CUDA版本不是11.6，你可能需要在cu116部分做相应的调整，查看官网相应的版本匹配。

4 用conda安装

conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia

5 验证安装

在Python中运行以下代码，确保pytorch安装成功。

import torch
print(torch.__version__)

在Python中运行以下代码，确保PyTorch能够使用CUDA：

**自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。**

**深知大多数Python工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！**

**因此收集整理了一份《2024年Python开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。**

![img](https://img-blog.csdnimg.cn/img_convert/348617144589fb3544d7423db15c77a9.png)

 

![img](https://img-blog.csdnimg.cn/img_convert/7cb95f010c730ea6af873e80f5dffb2c.png)

![img](https://img-blog.csdnimg.cn/img_convert/46506ae54be168b93cf63939786134ca.png)

![img](https://img-blog.csdnimg.cn/img_convert/252731a671c1fb70aad5355a2c5eeff0.png)

![img](https://img-blog.csdnimg.cn/img_convert/6c361282296f86381401c05e862fe4e9.png)

![img](https://img-blog.csdnimg.cn/img_convert/9f49b566129f47b8a67243c1008edf79.png)

 

**既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上前端开发知识点，真正体系化！**

**由于文件比较大，这里只是将部分目录大纲截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且后续会持续更新**

**如果你觉得这些内容对你有帮助，可以扫码获取！！！（备注Python）**

端开发知识点，真正体系化！**

**由于文件比较大，这里只是将部分目录大纲截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且后续会持续更新**

**如果你觉得这些内容对你有帮助，可以扫码获取！！！（备注Python）**

<img src="https://img-community.csdnimg.cn/images/fd6ebf0d450a4dbea7428752dc7ffd34.jpg" alt="img" style="zoom:50%;" />