pytorch-gpu
- 1. 确定cuda版本
- 2. 确定Python版本
- 3 开始下载-cu118-cp38
- 3.1 下载cuda
- 3.2 下载torchvision
- 4.下载好了
- 5.开始安装
- 6. 开始验证
1. 确定cuda版本
nvcc -V
版本为11.8 , 一会下载的版本为cu118
2. 确定Python版本
确定python版本为为3.8,一会下载为cp38
3 开始下载-cu118-cp38
https://download.pytorch.org/whl/torch_stable.html
(我测试了几个版本,发现只要保证cu118, cp38就可以。至于中间的torch版本,torchvision版本。什么版本都可以,因此你只要对应上cuda和python的版本就行)
3.1 下载cuda
3.2 下载torchvision
4.下载好了
5.开始安装
pip install “torch-2.0.0+cu118-cp38-cp38-win_amd64.whl”
pip install “torchvision-0.15.0+cu118-cp38-cp38-win_amd64.whl”
6. 开始验证
python
import torch
print(torch.cuda.is_available())
显示True,表示安装成功!!