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

前言

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

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

目录

一、PyTorch官网

二、查看安装方式

1）安装稳定版本

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

三、安装PyTorc

四、查看安装情况

一、PyTorch官网

https://pytorch.org/

二、查看安装方式

1）安装稳定版本

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

比如：在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/

三、安装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

安装完成啦～

四、查看安装情况

能看到成功安装pytorch1.8.0啦

欢迎交流～

文章来源: guo-pu.blog.csdn.net，作者：一颗小树x，版权归原作者所有，如需转载，请联系作者。

原文链接：guo-pu.blog.csdn.net/article/details/114761958