首页 > python教程

解决安装pytorch因网速问题失败的情况

时间:2021-07-13 python教程 查看: 1078

前言

每次从官网提供的下载方式,使用conda安装pytorch的时候都会因为网速问题导致失败。

使用conda清华源安装也不失为一种解决方案,然而这种方式在指定cuda版本时非常不方便。

本文主要针对这一问题给出解决方案。

方法

以安装pytorh-nightly, cuda80, linux系统为例。

pytorch官网选择 pip安装,并选定对应的系统, cuda版本, python版本。这时,会出现两条对应的安装命令。

第一条为安装numpy和torchvison的命令,可直接执行。

第二条为安装pytorch对应的命令,直接执行安装速度很慢。这里我们可以采用预先下载安装包的方式安装pytorch。

方法如下:

在命令上输入pytorch显示的第二条命令

pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cu80/torch_nightly.html

命令行会显示如下内容

其中

https://download.pytorch.org/whl/nightly/cu80/torch_nightly-1.1.0.dev20190429-cp37-cp37m-linux_x86_64.whl

是我们需要提前下载的安装包地址。

通过迅雷下载安装包。

将安装包上传至服务器

使用pip 离线安装pytorch包

source activate python虚拟环境名
pip install /path/of/the/pytorch_package(pytorch离线安装包路径)

至此,大功告成!

补充:ubuntu中利用anaconda快速配置pytorch,及网络源连接失败问题解决

在ubuntu中利用anaconda可以方便快速的配置pytorch。

具体步骤如下:

1.安装anaconda

bash Anconda3-x.x.x-Linux-x86_64.sh

2.创建python虚拟环境,这里是python3.6,环境名自拟。

conda create -n <your-virtualenv-name> python=3.6

3.激活进入虚拟环境

source activate <your-virtualenv-name>

4.加入清华源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

5.安装pytorch,可以根据pytorch官网信息自行匹配版本,这里选择cuda10.1

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

6.检核pytorch是否安装成功可以在进入python环境之后,输入下面命令

import torch 
print(torch.__version__) 
#如果返回版本信息,则安装成功

PS:若在第5步安装过程中存在如下所示连接网络失败错误,则删去命令后面 -c python 再次执行

CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://conda.anaconda.org/pytorch/linux-64/pytorch-1.3.1-py3.6_cuda10.1.243_cudnn7.6.3_0.tar.bz2>
Elapsed: -

An HTTP error occurred when trying to retrieve this URL.
HTTP errors are often intermittent, and a simple retry will get you on your way.

以上为个人经验,希望能给大家一个参考,也希望大家多多支持python博客。

展开全文
上一篇:Keras 如何修改图片通道的顺序
下一篇:教你利用Python破解ZIP或RAR文件密码
输入字:
相关知识
Python 实现图片色彩转换案例

我们在看动漫、影视作品中,当人物在回忆过程中,体现出来的画面一般都是黑白或者褐色的。本文将提供将图片色彩转为黑白或者褐色风格的案例详解,感兴趣的小伙伴可以了解一下。

python初学定义函数

这篇文章主要为大家介绍了python的定义函数,具有一定的参考价值,感兴趣的小伙伴们可以参考一下,希望能够给你带来帮助,希望能够给你带来帮助

图文详解Python如何导入自己编写的py文件

有时候自己写了一个py文件,想要把它导入到另一个py文件里面,所以下面这篇文章主要给大家介绍了关于Python如何导入自己编写的py文件的相关资料,需要的朋友可以参考下

python二分法查找实例代码

二分算法是一种效率比较高的查找算法,其输入的是一个有序的元素列表,如果查找元素包含在列表中,二分查找返回其位置,否则返回NONE,下面这篇文章主要给大家介绍了关于python二分法查找的相关资料,需要的朋友可以参考下