GPU版PyTorch安装

最新推荐文章于 2025-10-12 23:36:55 发布

原创最新推荐文章于 2025-10-12 23:36:55 发布 · 4.1w 阅读

20 ·

CC 4.0 BY-SA版权

深度学习与计算机视觉专栏收录该内容

2 篇文章

订阅专栏

本文提供了一种在Linux环境下使用Anaconda安装PyTorch的方法，并通过设置清华TUNA镜像源加速了安装过程。文章详细介绍了如何配置conda以使用镜像源，并给出了具体的安装命令。

登录PyTorch官网：http://pytorch.org/
这里写图片描述
利用conda（官网推荐用Anaconda安装）安装Linux版的的PyTorch，选择Python2.7， CUDA 8版本的。安装的时候就一条命令conda install pytorch torchvision -c pytorch,发现特别慢（墙的原因），所以参考这位大神的博客。
用清华TUNA镜像源，包含Anaconda仓库的镜像，将其加入conda的配置，配置如下，在终端分别输入以下两条命令：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --set show_channel_urls yes

同时需要在~/.condarc中把- defaults那行删除，这样每次安装新的软件包就会从TUNA镜像下载。然后再输入安装命令即可（conda install pytorch torchvision cuda80 -c soumith）。选择用conda安装linux,Cuda版本：cuda8.0,Python版本：python2.7。
安装成功：
这里写图片描述
推荐PyTorch中文网：https://ptorch.com/