1.看自己的CUDA,在cmd输入nvidia-smi
2.下载torch
download.pytorch.org/whl/torch_stable.html
- 选择我的cuda版本的10.1
- torch-1.4.0:表示torch版本为1.4.0
- cp38:表示适用python版本为3.8
- win:表示适用于win系统
- amd_64:表示64位系统
3.在python解释器的Scripts文件放whl文件
1.看自己的CUDA,在cmd输入nvidia-smi
2.下载torch
download.pytorch.org/whl/torch_stable.html
3.在python解释器的Scripts文件放whl文件