CODEWITHSUNDEEP
pythonpytorch
Spaceship222
Spaceship222

Reputation: 849

Does Pytorch 1.4 support cuda10.1?

I try to install pytorch 1.5+cu101, but it can't find caffe2_nvrtc.dll(it indeed exists in lib\caffe2_nvrtc.dll). So I try to install pytorch 1.4+cu101, but I can not find such wheel(only pytorch1.4+cu100/cu92) from https://download.pytorch.org/whl/torch_stable.html.
Any help?

Upvotes: 0

Views: 1503

Answers (1)

Michael Jungo
Michael Jungo

Reputation: 32982

PyTorch 1.4.0 shipped with CUDA 10.1 by default, so there is no separate package with the cu101 suffix, those are only for alternative versions. You just need to install the regular torch package:

pip install torch==1.4.0 -f https://download.pytorch.org/whl/torch_stable.html

Upvotes: 1

Related Questions