![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
暖i
这个作者很懒,什么都没留下…
展开
-
数据集读入
Dataset类代码实战 在配置好环境后 from torch.utils.data import Dataset from PIL import Image import os class MyData(Dataset): def __init__(self,root_dir,label_dir): self.root_dir = root_dir self.label_dir = label_dir self.path = os.path.jo原创 2022-03-17 22:28:39 · 260 阅读 · 0 评论 -
配置环境torch
pytorch 可以用pycharm 和jupyter 各有好处 GPU 主要是下载cuda对应的包,弄了很长时间,最后不知道是哪一步,最终配置上了gpu 官网下的要么很慢,要么根本不是gpu,是cpu,最后在官网上找的比较低版本的cuda,version,audio手动下载, 手动安装教程 jupyter就更气人了,在anaconda prompt手动安装,能打开,但是一直ln[*],服务器一直连接不上,还有什么runtime错误,网上说应该是版本不匹配,内核和什么 有的搞另外低,所以 C:\Users原创 2022-03-17 22:21:14 · 436 阅读 · 0 评论