Pytorch学习-Pytorch各版本安装详解

最新推荐文章于 2024-07-05 19:45:13 发布

AI研习图书馆

最新推荐文章于 2024-07-05 19:45:13 发布

阅读量442

点赞数 3

分类专栏： PyTorch学习文章标签： Pytorch安装

本文链接：https://blog.csdn.net/weixin_44222014/article/details/103499606

版权

本文提供Pytorch在Windows、Linux和Mac上的安装教程，包括conda和pip两种方法。详细讲解每个步骤，助力深度学习学习者顺利安装Pytorch。

摘要由CSDN通过智能技术生成

个人微信公众号：AI研习图书馆，欢迎关注~

深度学习知识及资源分享，学习交流，共同进步~

1. 前言

本文主要简要介绍Pytorch命令行安装教程~

2. windows版本

conda安装

1.cuda9.0  python3.6/3.5/3.7
conda install pytorch -c pytorch
pip3 install torchvision
2.cuda8.0 python3.6/3.5/3.7
conda install pytorch cuda80 -c pytorch
pip3 install torchvision

Pip安装

1.CUDA8.0  python3.5
pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp35-cp35m-win_amd64.whl
pip3 install torchvision
2.CUDA8.0  python3.6
pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp36-cp36m-win_amd64.whl
pip3 install torchvision
3.CUDA8.0  python3.7
pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp37-cp37m-win_amd64.whl
pip3 install torchvision

1.CUDA9.0  python3.7
pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp37-cp37m-win_