从零开始学Pytorch(零)之安装Pytorch

简介: 从零开始学Pytorch(零)之安装Pytorch

Pytorch优势


  聊聊为什么使用Pytorch,个人觉得Pytorch比Tensorflow对新手更为友善,而且现在Pytorch在学术界使用的得更多,大有逆袭Tensorflow之势。最近两年的顶会文章中,代码用Pytorch的比Tensorflow多。大家如果对Tensorflow也感兴趣,完全可以学习了Pytorch之后继续学习Tensorflow,基本原理都是相通的。让我们开始开启愉快的Pytorch学习之旅吧!

在Ubuntu系统或是windows系统安装Pytorch


  首先打开Pytorch的官网:https://pytorch.org/。在首页我们可以看到有各种配置可选,我们这里选择CPU或是GPU的版本都可以。一般选择GPU版本的Pytorch,这样运行大型的程序速度会快很多。而要运行GPU的Pytorch,就需要在电脑(不管是Ubuntu系统还是windows系统)安装相应的CUDA9和cudnn7,这两个安装起来有时候会遇到一大堆问题,所以需要一步一步解决。推荐一个我写的在Ubuntu系统安装教程的博客,按照博客上写的一步步来就可以啦。博客链接:https://blog.csdn.net/xiewenrui1996/article/details/102736238

  其实还有一个更简洁的安装方法,就是安装anaconda,然后在anaconda创建虚拟环境,在虚拟环境里面安装GPU版的Pytorch,系统就会自动给你安装相应的CUDA和cudnn,这样就节省了一大半的时间。写程序的编译器推荐用Pycharm,里面集成了许多写程序需要的模块,简单实用。Pycharm可以跟在anaconda里面创建的GPU版的Pytorch相连起来,把创建的虚拟环境在Pycharm使用,具体操作为在编译器里面添加anaconda的路径就可以了。

接下来更新的内容


  公众号接下来会更新从零开始学Pytorch的系列内容,比如线性回归的实现、神经网络框架代码实现、GAN等等,欢迎持续关注!

参考文献


[1]《动手深度学习》李沐

相关实践学习
部署Stable Diffusion玩转AI绘画(GPU云服务器)
本实验通过在ECS上从零开始部署Stable Diffusion来进行AI绘画创作,开启AIGC盲盒。
相关文章
|
5月前
|
并行计算 PyTorch 算法框架/工具
anaconda+pytorch+opencv安装及环境配置
anaconda+pytorch+opencv安装及环境配置
407 0
|
机器学习/深度学习 PyTorch 算法框架/工具
Pytorch学习笔记(1):基本概念、安装、张量操作、逻辑回归
Pytorch学习笔记(1):基本概念、安装、张量操作、逻辑回归
293 0
Pytorch学习笔记(1):基本概念、安装、张量操作、逻辑回归
|
5月前
|
PyTorch 算法框架/工具 Python
pytorch安装教程
pytorch安装教程
124 0
|
4月前
|
PyTorch 算法框架/工具
win10下安装pytorch,torchvision遇到的bug
win10下安装pytorch,torchvision遇到的bug
33 1
|
3天前
|
并行计算 开发工具 异构计算
在Windows平台使用源码编译和安装PyTorch3D指定版本
【10月更文挑战第6天】在 Windows 平台上,编译和安装指定版本的 PyTorch3D 需要先安装 Python、Visual Studio Build Tools 和 CUDA(如有需要),然后通过 Git 获取源码。建议创建虚拟环境以隔离依赖,并使用 `pip` 安装所需库。最后,在源码目录下运行 `python setup.py install` 进行编译和安装。完成后即可在 Python 中导入 PyTorch3D 使用。
|
2月前
|
机器学习/深度学习 人工智能 PyTorch
【Deepin 20深度探索】一键解锁Linux深度学习潜能:从零开始安装Pytorch,驾驭AI未来从Deepin出发!
【8月更文挑战第2天】随着人工智能的迅猛发展,深度学习框架Pytorch已成为科研与工业界的必备工具。Deepin 20作为优秀的国产Linux发行版,凭借其流畅的用户体验和丰富的软件生态,为深度学习爱好者提供理想开发平台。本文引导您在Deepin 20上安装Pytorch,享受Linux下的深度学习之旅。
70 12
|
2月前
|
PyTorch 算法框架/工具 Python
安装anaconda配置pytorch虚拟环境遇到的问题及解决办法
本文介绍了在配置Anaconda时遇到`PackagesNotFoundError`的问题,并提供了通过添加`conda-forge`通道和创建指定Python版本的PyTorch虚拟环境来解决这个问题的方法。
|
2月前
|
并行计算 Ubuntu PyTorch
【Deepin 20系统】Linux系统Anaconda安装Pytorch
本文提供了在Deepin 20系统上使用Anaconda安装Pytorch的详细步骤,包括创建一个新的conda环境、检查系统Cuda版本、选择对应Cuda版本的Pytorch安装命令、添加镜像源以加速下载,以及执行安装命令。
45 1
|
2月前
|
PyTorch 算法框架/工具
Pytorch 安装
Pytorch 安装
29 0
|
4月前
|
机器学习/深度学习 并行计算 PyTorch
安装PyTorch详细步骤
安装PyTorch时,选择CPU或GPU版本。有Nvidia显卡需装CUDA和cuDNN,可从NVIDIA官网下载CUDA 11.8和对应版本cuDNN。无Nvidia显卡则安装CPU版。安装PyTorch通过conda或pip,GPU版指定`cu118`或`rocm5.4.2`镜像源。验证安装成功使用`torch._version_`和`torch.cuda.is_available()`。