比如1.7.1, cuda 10.1
pip install torch==1.7.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html
官网查看有哪些可以装的:
https://download.pytorch.org/whl/torch_stable.html
比如1.7.1, cuda 10.1
pip install torch==1.7.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html
官网查看有哪些可以装的:
https://download.pytorch.org/whl/torch_stable.html