torch-torchvision-python quick install

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
insert image description here
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! !
insert image description here
insert image description here
Once you find the file, download and install it.

3 installation

pip install 路径+文件名

Guess you like

Origin blog.csdn.net/lafsca5/article/details/129698199