Tips:不建议conda直接下载,速度极慢
! ! It is recommended to download the wheel locally and then install it! !
1 Pay attention to version matching
Link: https://github.com/pytorch/vision#installation
Note that the version of cuda should also correspond! ! !
2 Download Wheel
https://download.pytorch.org/whl/torch_stable.html
Operation process:
For example, I need to install torch1.7.0
①Look up the table
torch1.7.0 corresponds to torch0.8.0 or 0.8.1
②Download
Enter the link to download the above two files, pay attention to the same cuda version! !
Once you find the file, download and install it.
3 installation
pip install 路径+文件名