os.environ['CUDA_VISIBILE_DEVICE']='4,5,6,7'
net = AttU_Net().to("cuda")
net = nn.DataParallel(net, device_ids=[0,1,2,3])
直接运行或者是在终端输入下面的命令：
CUDA_VISIBLE_DEVICES=4,5,6,7 python test.py
os.environ['CUDA_VISIBILE_DEVICE']='4,5,6,7'
net = AttU_Net().to("cuda")
net = nn.DataParallel(net, device_ids=[0,1,2,3])
直接运行或者是在终端输入下面的命令：
CUDA_VISIBLE_DEVICES=4,5,6,7 python test.py
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
