PyTorch在Windows,Linux,Mac安装方法

本文详细介绍了在Windows、Linux和Mac操作系统上安装PyTorch的步骤,包括使用conda和pip两种方法。对于Mac用户,由于二进制文件不支持CUDA,因此若需CUDA支持需从源代码安装。
摘要由CSDN通过智能技术生成

1、windows安装Pytorch

  • conda安装法
Python版本 CUDA版本 指令
Python3.5/3.6/3.7 CUDA8.0 1、conda install pytorch cuda80 -c pytorch 2、pip3 install torchvision
Python3.5/3.6/3.7 CUDA9.0 1、conda install pytorch -c pytorch 2、pip3 install torchvision
Python3.5/3.6/3.7 CUDA9.2 1、conda install pytorch cuda92 -c pytorch 2、pip3 install torchvision
Python3.5/3.6/3.7 None(即CPU环境) 1、conda install pytorch-cpu -c pytorch 2、pip3 install torchvision
  • pip安装法
Python版本 CUDA版本 指令
Python3.5 CUDA8.0 1、pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp35-cp35m-win_amd64.whl 2、pip3 install torchvision
Python3.6 CUDA8.0 1、pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp36-cp36m-win_amd64.whl 2、pip3 install torchvision
Python3.7 CUDA8.0 1、pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp37-cp37m-win_amd64.whl<
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值