- 博客(2)
- 收藏
- 关注
RSS订阅原创 cuda安装成功但是torch.cuda.is_available()为False的问题
安装好cuda之后在pycharm中print(torch,cuda,is_available())为False，经过查找之后发现是cuda版本和torch以及torchvision版本不对应导致的。我的cuda版本是12.1，过程供大家参考。然后下载对应版本的torch，直接卸载torch全家桶。
2023-04-30 10:37:46
831
1
原创 深度学习代码写作顺序
train：构建模型的训练策略以及评估方法 ，先构建前面那俩能够添加限制减少train的复杂度，在这部分，需要确定模型的更新策略、训练总时长、优化器类型、学习率大小和衰减策略、参数初始化方法、模型损失函数。datasets确定好model之后根据对应的输入输出格式才能确定对应的输出格式。需要考虑数据的存储位置和存储方式，pl：是不是分布式存储？是否存在多机多卡的情况？什么是model，datasets，train，什么是他们背后的逻辑和内容。model确定了神经网络的输入格式和输出格式。
2023-04-17 08:48:01
69
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人