ubuntu安装pytorch
?第一次在服务器上安装gpu版的pytorch,真的是一波三折,反反复复,搞了一个下午才安装成功。 ?记录一下安装过程中出现的错误,以及解决办法,也希望可以帮助到大家。
0、前置工作
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则表示安装成功。
4、同样可用于 torchtext、torchaudio等安装
|