利用 Conda 安装深度学习框架 Pytorch
1. 前言
Pytorch 目前是炙手可热的深度学习框架。和 TensorFlow 比较起来学习曲线更加平滑,不用写大量的样板代码就可以对网络进行训练和使用。在新版本的 Pytorch 中开始支持 Java 。但是安装 Pytorch 并不是很容易的事。今天就来说一下如何利用 Conda 安装 Pytorch 。
2. Conda
这里简单提一下 Conda ,Conda 是一个开源的、跨平台的软件包管理系统和环境管理系统,用于安装多个 Python 版本的软件包及其依赖关系,并在它们之间轻松切换。你可以把 Conda 看作 Python 领域的 Maven ,当然有些功能可能比 Maven 更加强大。通常我会安装 anaconda 来集成 Conda 和 Python 环境,anacoda 额外还有 numpy 、pandas 等有用的科学计算包。如果你有洁癖也可以使用 MiniConda ,它只包含 Conda 和 Python 。如果你还不会 Conda 建议你花十分钟快速入门一下。
3. 在线安装Pytorch
以 Mac 为例,Conda 安装新版 Pytorch 的命令如下:
conda install pytorch torchvision -c pytorch
相关文章