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

(完)