Ubuntu18 搭建PyTorch开发环境（Conda版）

末蓝、 2022-11-08 14:17 318阅读 0赞

## 前言 ##

在Ubuntu18通过Conda搭建PyTorch开发环境，记录一下安装过程，分享给大家参考。

如果没有安装Conda的朋友，可以参考一下这篇文章：[Ubuntu18安装Anaconda3（Conda）][Ubuntu18_Anaconda3_Conda]

**目录**

一、PyTorch官网

二、查看安装方式

1）安装稳定版本

2）安装旧版本或其他版本

三、安装PyTorc

四、查看安装情况

--------------------

## 一、PyTorch官网 ##

[https://pytorch.org/][https_pytorch.org]

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQxMjA0NDY0_size_16_color_FFFFFF_t_70][]

## 二、查看安装方式 ##

### 1）安装稳定版本 ###

在PyTorch官网首页拉到底下，能看到不同的安装方式啦；网上有好多文章介绍如何安装，但感觉安装过程繁琐，容易出错；参考一下官方的安装方式多好呀，简洁高效～

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQxMjA0NDY0_size_16_color_FFFFFF_t_70 1][]

比如：在ubuntu18中就选择Linux系统，我们使用Conda方式，选择CUDA的版本（有最新的，也有旧版本）

当然可以选择pip的方式安装，但感觉有时比较慢一些。

我们只需复制官方给出的命令进行安装即可，参考上图：**conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch**

### 2）安装旧版本或其他版本 ###

有时候项目环境要求，需要安装指定版本的PyTorch、CUDA，这时只需点击**Previous versions of PyTorch**

来到这个页面，查找合适的版本进行安装即可。

[https://pytorch.org/get-started/previous-versions/][https_pytorch.org_get-started_previous-versions]

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQxMjA0NDY0_size_16_color_FFFFFF_t_70 2][]

## 三、安装PyTorc ##

进入一个Conda环境，然后执行官方的安装命令就可以啦：

> 或者新建一个Conda环境，制定使用python3.8版本，conda create  -n Pytorch1.8  python=3.8
> 
> 进入刚才创建的环境：conda activate  Pytorch1.8   ；然后执行官方的安装命令就可以啦：

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQxMjA0NDY0_size_16_color_FFFFFF_t_70 3][]

安装完成啦～

## 四、查看安装情况 ##

能看到成功安装pytorch1.8.0啦

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQxMjA0NDY0_size_16_color_FFFFFF_t_70 4][]

欢迎交流～

[Ubuntu18_Anaconda3_Conda]: https://guo-pu.blog.csdn.net/article/details/114532230
[https_pytorch.org]: https://pytorch.org/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQxMjA0NDY0_size_16_color_FFFFFF_t_70]: /images/20221023/94f41dd738ac41b2be50ceef597dc269.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQxMjA0NDY0_size_16_color_FFFFFF_t_70 1]: /images/20221023/d089c97d484a4a4a8bd30d789c5fcde2.png
[https_pytorch.org_get-started_previous-versions]: https://pytorch.org/get-started/previous-versions/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQxMjA0NDY0_size_16_color_FFFFFF_t_70 2]: /images/20221023/83b12c393118447bbd059eded6ba3a1f.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQxMjA0NDY0_size_16_color_FFFFFF_t_70 3]: /images/20221023/b027729ad62546ed948c968614bb1fad.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQxMjA0NDY0_size_16_color_FFFFFF_t_70 4]: /images/20221023/a674baf1c4c14ffa905d383d20ac6302.png