常规安装
-
进到pytorch官网:https://pytorch.org/
-
选择pip或者conda安装
由于网络等原因安装失败
Could not fetch URL https://pypi.org/simple/torch/:
There was a problem confirming the ssl certificate: HTTPSConnectionPool(host='pypi.org', port=443): Max retries exceeded with url: /simple/torch/
(Caused by SSLError(SSLZeroReturnError(6, 'TLS/SSL connection has been closed (EOF) (_ssl.c:1131)'))) - skipping
这里采用pip安装方式,见torch官网中pip后面指定的参数--extra-index-url https://download.pytorch.org/whl/cu116
,意思是除了在pip默认的安装网站https://pypi.org/simple
寻找torch包以外,还会在以上指定的额外url里寻找torch包,见如下代码:
(torchgpu) C:\Users\xxx>pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
Looking in indexes: https://pypi.org/simple, https://download.pytorch.org/whl/cu116
因为pip官网包很多,寻找指定torch很花时间,这里直接指定网站下载,将--extra-index-url
参数换成指定网站下载-i
,此时结果如下:
(torchgpu) C:\Users\xxx>pip3 install torch torchvision torchaudio -i https://download.pytorch.org/whl/cu116
Looking in indexes: https://download.pytorch.org/whl/cu116
Collecting torch
Downloading https://download.pytorch.org/whl/cu116/torch-1.13.1%2Bcu116-cp38-cp38-win_amd64.whl (2434.2 MB)
------ --------------------------------- 0.4/2.4 GB 3.2 MB/s eta 0:10:42
官网下载whl包进行本地安装
可见能够在这个网站上下载,但是后续下载一半又出现erro,无法下载完全,这时候直接进到上述网站:https://download.pytorch.org/whl/cu116/torch-1.13.1%2Bcu116-cp38-cp38-win_amd64.whl
,在浏览器打开会自动下载该whl包,此时能够顺利下载,下载完成后cd到whl文件所在文件夹,利用pip指定安装包进行安装(以下两种应该都可以):
pip install whl "torch-1.13.1+cu116-cp38-cp38-win_amd64.whl"
pip install torch-1.13.1+cu116-cp38-cp38-win_amd64.whl
注:
后续发现可以直接进到pytorch指定pip下载wheel网站:https://download.pytorch.org/whl/cu116
根据指令,下载符合要求的torchvision torchaudio pytorch-cuda=11.6
,分别查找符合cuda版本的三个文件
点进torch,利用网页查找功能(ctrl+F),图中圈出四个参数分别是torch版本、cuda版本、python版本、操作系统版本;
成功安装好最大的torch后,可继续使用pip安装其余两个,因为剩余两个比较小,一般pip安装不会中断。