深度学习记录
记录在加载模型、建模、训练过程中遇到的问题
音无八重
懒宅
展开
-
CUDA查找与选用GPU设备
device()来设置数据载入的显卡,这个1号显卡的名称为“cuda:0”,而不是“cuda:1”,因为它是可使用显卡中的第0位,如果设置成“cuda:1”就会发生。在第1点中你查看可使用显卡个数的话,显卡个数为3。一般情况下是默认第0张显卡为主卡,如果你想设置其他显卡为主卡的话,可以使用。torch.cuda.is_available() #是否有效。torch.cuda.device_count() #个数。在显卡上加载及运算时,就会按照1->2->0的顺序加载。原创 2024-05-10 15:09:06 · 340 阅读 · 0 评论 -
随笔记录
本来前几天还想要写一下神经网络的例子的,奈何最近心情起伏过大,正在准备考研的调剂,等过几天有精力了再写吧。积攒未写的博客:(1)有关神经网络的具体而简单的例子;(2)有关kaggle中Titanic项目的学习流程与经验...原创 2020-05-17 09:18:28 · 239 阅读 · 0 评论