摘要: # PyTorch查看模型和数据是否在GPU上 data = torch.zeros([1,2]) print(data.device) #此外也可以用.is_cuda也可以判断模型和数据是否在GPU上,例如:data.is_cuda 1.查看tensor所在的设备 data = data.cuda 阅读全文
posted @ 2023-01-06 10:12 萧泽 阅读(2634) 评论(0) 推荐(0)
摘要: 1.扩大Pycharm运行内存 打开后找到-Xms -Xmx两行,增加运行内存(根据电脑配置),可明显改善打开IDE的速度 我刚开始配置-Xms为2048m,-Xmx为4096m,会遇到打不开的情况,这时候减小为-Xms为1024,-Xmx为2048就可以顺利打开了 2.解决运行时查看变量速度慢的方 阅读全文
posted @ 2023-01-06 09:33 萧泽 阅读(2462) 评论(0) 推荐(1)