anaconda3在win10环境下安装pytorch0.4 + cuda9.0

前言
一、安装torch与torchvision
初次安装的时候,没有经验,在安装CUDA时踩了很多坑,各种帖子都看了,比如去官网下载最新的驱动,用DDU将电脑上的显卡卸载干净等,最后还是没装上CUDA。果断放弃了。
最后看了各种资料突然发现 ,原来使用 conda install pytorch -c pytorch 命令安装pytorch时,就默认安装了CUDA!!!天了噜,折腾了那么多时间!不过也各有利弊吧!
默认安装的是cuda8.0版本,若安装其他版本的cuda,请用参数指定即可,比如我安装cuda9.0 就使用命令 conda install pytorch cuda90 -c pytorch 即可,其他版本的也类似!
上述方式仅是安装了torch模块(实现神经网络需要的核心模块),但是我们仍然需要一个辅助模块,帮助我们做一些数据处理、数据库加载、视觉图像处理等任务,那就是torchvision模块。 安装命令为 conda install torchvision -c pytorch ,执行这个命令安装的时候,可能会报 PackagesNotFoundError: The following packages are not available from current channels的错。这个时候换pip方式安装就行了,执行pip install torchvision即可安装。
二、验证安装
首先进入python的交互环境,执行如下操作,如果打印出模块安装的位置了,即表明安装成功。(当然也有其他验证方式,比如用版本号等,这里不赘述)

>>> import torch
>>> print (torch)
<module 'torch' from 'D:\\project\\lib\\site-packages\\torch\\__init__.py'>
>>> import torchvision
>>> print (torchvision)
<module 'torchvision' from 'D:\\project\\lib\\site-packages\\torchvision\\__init__.py'>

三、关于cudann的问题
很多小伙伴可能还不明白cuda与cudann的区别,这里简单说一下。
cuda是英伟达推出的一个并行计算平台,即所谓的GPU加速计算,它是一个框架模型。
cudann呢,是专门针对深度神经网络推出的一个加速库,用来提升模型的训练速度等,不是一个必须的库。
博主的电脑上面暂时没有按照cudann,一是因为暂时来说,我还用不到它;二是尝试安装了一下,在VS2017上面验证的时候,遇到了非cudann的bug. 心累就不搞了,以后真正用到了,再更新安装!
学习的过程中遇到了两篇感觉不错的博客,这里分享一下。
pytorch安装 :https://blog.csdn.net/xiangxianghehe/article/details/80103095
torchvision安装:https://blog.csdn.net/Tan915730/article/details/78954482

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值