Pycharm搭建CUDA，Pytorch教程（匹配版本(1)

最新推荐文章于 2024-07-28 17:29:20 发布

2401_83621384

最新推荐文章于 2024-07-28 17:29:20 发布

阅读量155

点赞数 3

分类专栏： 2024年程序员学习文章标签： pycharm pytorch ide

本文链接：https://blog.csdn.net/2401_83621384/article/details/137357921

版权

本文详细介绍了如何在PyCharm中设置CUDA环境，以便使用PyTorch进行深度学习计算。通过匹配合适的CUDA版本，实现GPU加速的PyTorch项目开发。

摘要由CSDN通过智能技术生成

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，1.6.0，1.5.1，1.5.0 |
| 11.0 | 1.7.1，1.7.0 |
| 11.1 | 1.8.0 |
| 11.3 | 1.12.1，1.12.0，