windows10下装pytorch简单步骤和中遇见的一些问题

首先分享一个博客的链接,我开始是在这上面学习的,之后遇见的一些问题,就自己找方法解决的。

基本教程,按着这上面来基本错不了

还有一个链接也是安装教程 链接

我是自己想在windows10上面装pytorch,因为自己在虚拟机上装了ubuntu16.04,配好了tensorflow的环境。

我自己是先装好了python3.6并且电脑带有GPU的 是 grforce 940MX

查看自己是否安装在cmd nvcc -V

首先要安装显卡驱动cuda 建议安装cuda9.1版本,开始在官网找不到这个安装包不过还是找到了现在最新是cuda10.0版本

附上链接吧下载地址

下载之后就可以开始安装了

  1. 首先点开后,选择自定义安装,会出现安装路径,可以自己设置,但是不重要安装后自己就会消失。

2.安装时可以看到,cuda中的Visual Studio Integration不要勾选,其他的勾上,默认安装路径,然后开始安装。

3.环境变量的配置

4.查看安装是否成功

cmd后  set cuda

5.进入cuda的安装路径

以上没问题的话,就是cuda9.1安装成功了。

还要加上cudnn

下载地址官网

接下来安装pytorch吧

选择pip安装

在cmd输入

  1. pip install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl

开始安装,这里也是需要翻墙的速度快,不然很慢。

之后安装torchvision,通过本地安装

下载地址    地址

让后在下载后的文件夹命令安装

  1. pip install torchvision-0.2.1-py2.py3-none-any.whl

最后看torch是否可以使用

在这里遇到一个问题,import torch失败

最后再这篇文章找到了,解决方法

主要是环境变量配好,配到路径中去。

地址,,可能需要翻墙

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值