mei/blog
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 维护一下远古文章们

Browse Source
This commit is contained in:
mei
2026-04-17 22:33:46 +08:00
parent fe8f71dd4e
commit e2b65d15c3
41 changed files with 742 additions and 386 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/_posts/2024/02/cuda122usepytorch.md
+10 -4
View File
@@ -11,18 +11,24 @@ banner_img: /img/2024/2/17/torch.png
permalink: /articles/2024/cuda122-use-pytorch.html
date: 2024-02-17 11:36:55
---
如果你在nvidia官网安装了最新版的CUDA(CUDA12.2/12.3),但访问 [https://download.pytorch.org/whl/cu122/torch_stable.html](https://download.pytorch.org/whl/cu122/torch_stable.html) 就会发现pytorch并没有支持此版本的CUDA
但是支持CUDA12.1的pytorch也可以支持CUDA12.2(看网上说CUDA12.3也可以支持,但没有试过)
执行
如果你在nvidia官网安装了最新版的CUDA(CUDA12.2/12.3),但访问 [https://download.pytorch.org/whl/cu122/torch_stable.html](https://download.pytorch.org/whl/cu122/torch_stable.html) 就会发现pytorch并没有支持此版本的CUDA
但是支持CUDA12.1的pytorch也可以支持CUDA12.2(看网上说CUDA12.3也可以支持,但没有试过)
执行
```shell
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu121/torch_stable.html
```
来安装支持CUDA12.1的torch
### 检查
创建一个.py文件,内容为:
```python
import torch
print(torch.cuda.is_available())
```
运行它,如果返回True,则说明PyTorch已成功启用CUDA支持
如果返回False,那么重装一下torch或重启一下服务器应该就好了
如果返回False,那么重装一下torch或重启一下服务器应该就好了