Pycharm搭建CUDA，Pytorch教程（匹配版本，安装，搭建全保姆教程）_cuda12(1)

最新推荐文章于 2025-04-05 20:15:23 发布

2401_84557821

最新推荐文章于 2025-04-05 20:15:23 发布

阅读量624

点赞数 4

分类专栏：程序员文章标签： pycharm pytorch ide

本文链接：https://blog.csdn.net/2401_84557821/article/details/138393456

版权


查看cuda版本![](https://img-blog.csdnimg.cn/direct/ba9a3507f82a43b8891b42ab1566a71c.png)


输入

set cuda


查看环境变量![](https://img-blog.csdnimg.cn/direct/722e47935a14465a906ed5b923d3cd45.png)


如上两图即为下载成功！


## 二、安装Pytorch


### 1、了解对应的pytorch版本


要成功运行cuda架构，所需的pytorch版本必须与python和cuda版本对应，以下为cuda与pytorch对应关系





 cuda与pytorch版本对应表 
 | CUDA版本 | 可用PyTorch版本 |
| --- | --- |
| 7.5 | 0.4.1 ，0.3.0， 0.2.0，0.1.12-0.1.6 |
| 8.0 | 1.1.0，1.0.0 ，0.4.1 |
| 9.0 | 1.1.0，1.0.1, 1.0.0，0.4.1 |
| 9.2 | 1.7.1，1.7.0，1.6.0，1.5.1，1.5.0，1.4.0，1.2.0，0.4.1 |
| 10.0 | 1.2.0，1.1.0，1.0.1 ,1.0.0 |
| 10.1 | 1.7.1，1.7.0，1.6.0，1.5.1，1.5.0， 1.4.0，1.3.0 |
| 10.2 | 1.12.1，1.12.0，1.11.0，1.10.1，1.10.0，1.9.1，1.9.0，1.8.1，1.8.0，1.7.1，1.7.0ÿ