- 博客(1)
- 收藏
- 关注
转载 Pytorch(多GPU训练)转载简单操作
单GPU 开始用过全局指定os.environ["CUDA_VISIBLE_DEVICES"] = "0"的方法,但是运行会爆奇怪的cudnn错误,后面改成下面写法运行通过 my_device="cuda:0" #这里的0可以根据需要改为指定的gpu id #对于模型 model=model.to(device=my_device) #对于数据 input_data=Variable(inp...
2019-11-25 14:15:51 249
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人