前言
在Ubuntu18通过Conda搭建PyTorch开发环境,记录一下安装过程,分享给大家参考。
如果没有安装Conda的朋友,可以参考一下这篇文章:Ubuntu18安装Anaconda3(Conda)
目录
一、PyTorch官网
二、查看安装方式
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