logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

pytorch深度学习一机多显卡训练设置,流程

最近在学习在服务器的ubuntu环境上配置用多个显卡训练,之前只用一个显卡训练实在是太慢了点先看看服务器上有几个显卡:nvidia-smi即可得到具体的显卡信息:每个显卡之前有对应的编号。然后得知自己服务器上总共有多少显卡后,插入以下代码:#一机多卡设置os.environ['CUDA_VISIBLE_DEVICES'] = '0,1,2,3'#所有的显卡,共计四块device_ids = [0,

#pytorch#ubuntu#深度学习 +2
暂无文章信息