【安装记录】cuda版本与pytorch安装
·
pytorch版本基于cuda版本和python版本确定,不同环境对版本有不同要求,记录一下历程。
一、CUDA
1. 版本查看
首先打开终端【win+R → 输入cmd → 打开】
1.1 驱动版本【电脑当前驱动下支持的最高版本】
指令 : nvidia-smi

右上角的CUDA Version表示,当前支持的最高cuda版本为12.5
1.2 运行版本【编译深度学习代码时的运行版本】
指令 : nvcc -V

后两行的11.3表示当前运行CUDA的版本,也是编译代码时用到的版本,这个版本不能超过上述的驱动版本(12.5)
2. CUDA&Cudnn安装
参考教程:https://blog.csdn.net/YYDS_WV/article/details/137825313
关于cudnn的网页链接:
9.0之前的版本: https://developer.nvidia.com/rdp/cudnn-archive
9.0之后的新版本: https://developer.nvidia.com/cudnn-archive
二、pytorch
版本参考链接:https://pytorch.org/get-started/previous-versions/
由于torch大都是捆绑出现的,稍微整理了一下相关的版本
| torch | torchvision | torchaudio | torchdata | Xformers | cuda |
|---|---|---|---|---|---|
| 2.5 | 0.20 | 2.5.0 | 0.9.0,0.10.0 | ||
| 2.4.1 | 0.19.1 | 2.4.1 | 0.0.28 | 12.4 | |
| 2.4 | 0.19 | 2.4.0 | 0.8.0 | ||
| 2.3 | 0.18 | 2.3.x | 0.0.27 | 11.8/12.1 | |
| 2.2 | 0.17 | 2.2.x | 0.0.26,0.0.25,0.0.24 | 11.8/12.1 | |
| 2.1 | 0.16 | 2.1.0 | 0.0.23 | ||
| 2.0.1 | 0.15.1 | 2.0.2 | 0.0.22 | ||
| 2.0.0 | 0.15.1 | 2.0.1 | 0.6.0 |
安装包的时候可以指定版本,并提供对应cuda版本的下载参考链接,以cuda11.8为例:
pip install torch\==2.3.1+cu118 torchvision\==0.18.1+cu118 torchaudio\==2.3.1+cu118 xformers\==0.0.27 torchdata==0.7.1 --extra-index-url https://download.pytorch.org/whl/cu118
其中torch分为cpu和cuda版本,如果不说明+cu118,很可能下载的是cpu版本
更多推荐
所有评论(0)