PyTorch介绍与安装

最新推荐文章于 2024-08-08 22:04:44 发布

Vic时代

最新推荐文章于 2024-08-08 22:04:44 发布

阅读量2k

点赞数

分类专栏：深度学习框架Torch 文章标签： PyTorch Torch Python 深度学习

本文链接：https://blog.csdn.net/VictoriaW/article/details/58108731

版权

22 篇文章 28 订阅

订阅专栏

Facebook的AI研究团队发布了Python工具包——PyTorch，专门用于GPU加速的深度学习编程，实现了在Python语言下使用Torch。

PyTorch的安装非常方便。在主页中选择对应的操作系统、包管理器、Python版本、CUDA版本信息后会提示安装命令。不完全组合如下：

pip install http://download.pytorch.org/whl/cu75/torch-0.1.12.post2-cp27-none-linux_x86_64.whl 
pip install torchvision

pip install http://download.pytorch.org/whl/cu80/torch-0.1.12.post2-cp27-none-linux_x86_64.whl 
pip install torchvision

pip install http://download.pytorch.org/whl/cu75/torch-0.1.12.post2-cp35-cp35m-linux_x86_64.whl 
pip install torchvision

pip install http://download.pytorch.org/whl/cu80/torch-0.1.12.post2-cp35-cp35m-linux_x86_64.whl 
pip install torchvision

sudo pip uninstall torch

关注

专栏目录