安装gpu版本pytorch教程(避坑)
时间: 2023-11-05 15:54:31 浏览: 218
win10快速安装pytorch gpu版本
5星 · 资源好评率100%
安装GPU版本的PyTorch可以按照以下步骤进行操作:
1. 首先,访问链接[https://download.pytorch.org/whl/torch_stable.html](https://download.pytorch.org/whl/torch_stable.html),这里可以找到PyTorch不同版本的安装包。
2. 根据你的需求,选择对应的GPU版本安装包。注意,文件名以"cu"开头的是GPU版本,而以"cpu"开头的是CPU版本。确保你选择了正确的版本。
3. 创建一个虚拟环境,然后使用以下命令安装PyTorch和torchvision:
```
pip install torch
pip install torchvision
```
4. 安装完成后,可以打开Python进行测试,验证安装是否成功。
阅读全文