Linux ubuntu安装pytorch

最新推荐文章于 2024-08-03 11:32:27 发布

te_ar

最新推荐文章于 2024-08-03 11:32:27 发布

阅读量5.2k

点赞数

分类专栏：深度学习文章标签： pytorch linux ubuntu

本文链接：https://blog.csdn.net/qq_44117953/article/details/120956289

版权

深度学习专栏收录该内容

1 篇文章 0 订阅

订阅专栏

本文详细记录了在Ubuntu服务器上安装PyTorch的曲折过程，涉及直接安装、国内镜像配置及本地下载安装，提供了解决常见问题的方法，适用于CUDA版本的安装和torchtext、torchaudio等库。

摘要由CSDN通过智能技术生成

文章目录

ubuntu安装pytorch

ubuntu安装pytorch

第一次在服务器上安装gpu版的pytorch，真的是一波三折，反反复复，搞了一个下午才安装成功。
记录一下安装过程中出现的错误，以及解决办法，也希望可以帮助到大家。

0、前置工作

在Anaconda中创建pytorch环境

conda create -n pytorch python=3.8

激活环境

source activate pytorch

1、直接安装

打开pytorch官网：
https://pytorch.org/
选择需要版本：
复制Command指令到命令行，执行
等待安装完成即可
缺点：安装很慢，有时候中间还会断。（windows还能科学上网，ubuntu上就不太会了。。。）

2、国内镜像安装

2.1、安装

执行 ls -a，可以看到有 .condarc 文件，可以直接进入进行镜像通道添加，或者如下执行指令
在命令行执行：

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/conda-forge/
conda config --set show_channel_urls yes

2.2、问题

但是接下来执行上述安装 pytorch 指令，会报 Http 错误，无法进行下载。

CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://mirrors.tuna.tsi...

在这里插入图片描述

在网上查了一下，有以下解决方案：

打开 .condarc 文件，删除channels中的 -defaults
若上述仍无效，将前面添加的清华镜像源中的 https 修改为 http

再执行 pytorch 的conda安装指令。
本人到这一步确实可以运行了，但我明明安装的应该是带有 cuda 的pytorch版本，之后在 python 中使用 torch.cuda.is_availabel() 返回却是false。
经过排查，似乎是清华镜像源中并没有该版本的cuda pytorch，所以就下载的cpu版本 pytorch。

3、本地安装

1、进入Anaconda资源包下载链接
https://conda.anaconda.org/pytorch/linux-64/
在这里插入图片描述
2、找到需要的资源包

可以直接通过浏览器下载，个人感觉用迅雷下载比较快。

3、复制到服务器 ~/anaconda3/pkgs 文件夹下

4、执行本地安装指令

conda install --use-local ./anaconda3/pkgs/pytorch-1.10.0-py3.8_cuda10.2_cudnn7.6.5_0.tar.bz2

这里可能会需要下载其他的依赖包，但相较于700多M的pytorch包来说已经小很多了。

在这里插入图片描述

5、测试

import torch
torch.cuda.is_available()

若返回True则表示安装成功。
在这里插入图片描述