关于pytorch的一些事儿

markdown新手写的第一篇文章,想记录一下这两三天弄pytorch的一些事情

缘由

由于任务的原因,需要使用深度学习框架,一开始是准备使用TensorFlow的,后来在项目负责人的引导下选择了Pytorch(新手友好型,貌似),这两三天一直在折腾它。今天下午终于是弄好了前期的东西。 写下来做个记录吧。

经历

  1. 初期的选择
    一开始因为想起有同学使用TensorFlow进行深度学习,加上某人比较懒,于是乎,开始看有关TF的东西,准备在小破机上搞一个小的TensorFlow模型,看着看着就没有了下文,中间发生了一些事情,这个事情就被偶落下了。后来还是生活驱使我不断向前啊,总负责人来询问进度。呵呵~~~~~~ 一脸生无可恋,幸亏是在疫情期间,只能口头问下,也就草率的想了下原本准备干的TF计划,将其告知负责人(心慌,。) 负责人还是挺好的,认为我一开始就搞TF有一定难度,这个项目也比较浅一点,时间也有点不足吧,于是推荐我去看看pytorch,于是乎,一系列的原因(懒加上浪),我便开始弄pytorch了
    2.中期的挣扎
    关于pytorch,一开始我是想要安装在Ubuntu中的(因为装了双系统),后来在快弄完的时候发现Ubuntu空间不足了,cuda有点儿大,Ubuntu本来空间就分的不是特别多(50G),以前还弄过几个小的项目,装过几个大一点的软件,这时候就有点奔溃了,休息一下吧。 放了自己一个中午的假去休息了。当天下午改变计划,就安装到自己的windows上吧,windows 一开始是不能够搭建pytorch的,在某一版之后就ok了。不知道是不是点背的原因,装cuda的时候没怎么注意,装完之后才显示出问题,NVIDIA驱动不支持cuda过高版本,要装的兄弟们得注意一下了。然后就是一顿卸载、重装操作,再将cudnn的东西拷贝过去,然后GPU版pytorch的前期基础也就准备好了,之后根据网上教程安装anaconda3、使用conda包管理下载pytorch,anaconda3确实不错,这里面的jupyter notebook一直是我挺想要装的,由于某些原因没有弄(懒),用来做调试是非常不错的选择。不过在弄conda pytorch的时候又踩坑里了,各种被墙加上偏远地区网络有点差,无奈一怒之下选择清华镜像源直接下载压缩包:

    清华镜像源地址:https://mirrors.tuna.tsinghua.edu.cn/

后面就是离线安装 这里插一句,推荐用conda虚拟环境来弄,毕竟windows东西多,怕以后各种环境问题,放在一个新的虚拟环境里面以后会省不少功夫哟。

3.后期的收尾
上面各种东西弄好之后就可以在命令行中各种飞驰了,BUT 开发还是需要在编辑器或者IDE中进行的,我一直是VScode 的忠实用户,SO 先选择虚拟env 位置文件夹,
通过命令行命令:

Ctrl + Shift + P

输入:选择器
选择自己在虚拟环境中设置的Python版本
之后再env文件下创一个文件夹,用来放置项目代码吧
至此,可以使用torch了
放一段测试torch的代码吧:

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

ngpu= 1
# Decide which device we want to run on
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3,3).cuda()) 

代码编辑的时候报了一个torch是not callable的错误,但是不影响运行,大概是vscode 的 pylint 对于torch不友好吧(开个玩笑,得在设置里面设置忽略)

结果

这里就 放一张结果图(嘻嘻)吧:
在这里插入图片描述
至此,一篇小结便完成了,哈哈,争取每天更新更新。(明天准备弄弄BERT)

子曰:“不患无位,患所以立;不患莫己知,求为可知也。” ——《论语·里仁》

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值