cuda11.2. pip安装pytorch
时间: 2023-06-30 09:20:30 浏览: 494
您可以按照以下步骤使用pip安装PyTorch并与CUDA 11.2一起使用:
1. 确认您已经安装了CUDA 11.2和相应的驱动程序。
2. 打开终端或命令提示符,并输入以下命令来安装PyTorch:
```
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu112/torch_stable.html
```
3. 如果您想安装特定版本的PyTorch,请在命令中添加版本号:
```
pip install torch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 -f https://download.pytorch.org/whl/cu112/torch_stable.html
```
4. 等待安装完成后,您可以通过在Python中导入PyTorch来确认是否正确安装:
```
import torch
print(torch.__version__)
```
如果输出的版本号与您安装的版本号相同,则说明已经成功安装PyTorch并与CUDA 11.2一起使用。
阅读全文