# 显卡云使用带有CUDA支持的pytorch

**URL:** <https://forum.rainyun.com/t/topic/6940>\
**Category:** 云服务器 RCS\
**Created:** [2024 年2 月 15 日 10:25 UTC](https://forum.rainyun.com/t/topic/6940 "2024-02-15T10:25:17Z")\
**Posts on this page:** 1\
**Page:** 1

<div class="post-metadata">

作者： ![mei](https://forum.rainyun.com/user_avatar/forum.rainyun.com/mei/32/8588_2.png) [@mei](https://forum.rainyun.com/u/mei)\
发布日期： [2024 年2 月 15 日 10:25 UTC](https://forum.rainyun.com/t/topic/6940/1 "2024-02-15T10:25:17Z")

</div>

显卡云自带CUDA版本为12.2   
  
但访问`https://download.pytorch.org/whl/cu122/torch_stable.html`就会发现pytorch并没有支持此版本的CUDA   
  
但是支持CUDA12.1的pytorch也可以支持CUDA12.2(网上说CUDA12.3也支持)  
执行

```python
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或重启一下服务器应该就好了