【更新时间2019.5.14】pytorch 1.1+win10 +python3.68+cuda10.0 安装

这篇博客详细介绍了如何在Windows 10环境下,使用Python 3.68和CUDA 10.0安装PyTorch 1.1版本。提供了CPU和GPU两种版本的pip安装步骤,并强调不要同时安装不同CUDA版本。文章最后提到安装成功后的检测方法,并邀请遇到问题的读者加入特定QQ群寻求帮助。
摘要由CSDN通过智能技术生成

pytorch 1.1+win10 +python3.67+cuda10.0 安装 

点击进入官网 https://pytorch.org/


pip安装

Python3.68+pip安装cpu版本

pip3 install https://download.pytorch.org/whl/cpu/torch-1.1.0-cp36-cp36m-win_amd64.whl
pip3 install torchvision

Python3.68+pip安装gpu版本 
目前gpu版本支持cuda8.0,cuda9.0和cuda10.0,请选择对应的版本下载安装,不要同时执行下面三个命令!喜欢conda的可以选择conda安装

 
pip3 install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-win_amd64.whl
pip3 install torchvision


检测是否成功!

import torch
print(torch.cuda.is_available())

如果返回True则可以使用,经测试可以正常使用

不得不说,比tensorflow 的安装方便了100倍!

还有问题的,可以加群264191384 注明(pytorch)

-WIN10+cuda10+pytorch+py3.68环境下,warpctc 编译不成功的解决办法

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值