2020年2月18日
指定GPU训练模型
摘要： 公司或者实验室当大家都共用一台服务器时，训练模型的时候如果不指定GPU，往往会冲突。 我们可以先用 nvidia-smi 查看有多少块GPU， 然后分两种方式指定GPU运行。 1、直接在终端运行时加入相关语句实现指定GPU的使用 CUDA_VISIBLE_DEVICES=0 python test. 阅读全文
posted @ 2020-02-18 17:18 理想几岁 阅读(6006) 评论(0) 推荐(0)