1.普遍问题
使用pip在安装pytorch的过程中最难受的问题就是:pip._vendor.urllib3.exceptions.ReadTimeoutError: HTTPSConnectionPool(host='download.pytorch.org', port=443): Read timed out.
有时候甚至都下载到百分之八十或者九十了最后报了一个这种问题。
2.安装pip安装pytorch方法
经过尝试我找到了一个方法就是先把pytorch的轮子用迅雷下载好之后然后再安装。
步骤如下:首先在官网中选择pip安装pytorch并复制指令
然后在控制台粘贴指令并回车,重点就是这里会显示下载链接
将Dowloading这个下载链接复制,然后去迅雷,点击新建,然后将下载链接复制到上面。
立即下载之后将文件放在项目文件夹中,或者其他路径也行(这里我是放在了项目文件夹下)。
然后再在控制台中输入:pip3 install 文件路径。
3.检测是否安装成功
回车之后就安装成功啦,可以使用如下代码检测是否安装成功以及CUDA,GPU是否可用
import torch
print(torch.__version__)
print(torch.cuda.is_available())
显示版本号以及返回true则为成功。