Deep Learning
叫我咸鱼就好
努力学习
展开
-
两行代码搞定Pytorch多GPU深度学习(DataParallel)
当单张显卡无法满足深度学习的训练要求时,可以尝试增加显卡数量。比如当我们想增加batch size,但单张显卡的显存不足时,就需要增加显卡。在Pytorch中调用多张显卡并没有多么麻烦,只需要修改以下部分:允许显示哪些显卡,在代码开头加入:import osos.environ['CUDA_VISIBLE_DEVICES'] = "0, 1, 2, 3"# 调用四张显卡,显卡编号分别为为0...原创 2020-03-29 14:35:25 · 1119 阅读 · 0 评论 -
错误解决Object arrays cannot be loaded when allow_pickle=False
在运行《Deep Learning with Python》书中的代码时出现了这一个错误。书内代码如下:from keras.datasets import imdb(train_data, train_labels), (test_data, test_labels) = imdb.load_data(num_words=10000)from keras.datasets import ...原创 2020-02-07 06:40:15 · 5475 阅读 · 0 评论