当前位置：网站首页>pytorch训练深度学习网络设置cuda指定的GPU可见

pytorch训练深度学习网络设置cuda指定的GPU可见

2022-07-01 03:15:00 【是七叔呀】

cuda指定的GPU可见

  • 原来的训练命令为：python train.py --config config/pascal/pascal_split0_resnet50.yaml ，但是会用到GPU上所有的显卡
  • 想使用指定显卡训练，可以在自己的python命令前加上：CUDA_VISIBLE_DEVICES=0,1，这里0，1是自己的显卡号
  • CUDA_VISIBLE_DEVICES=0,1 python train.py就可以用两张卡0，1正常训练了
    在这里插入图片描述

！！！注意这里CUDA_VISIBLE_DEVICES=0,1，这里CUDA_VISIBLE_DEVICES、等号和0，1之间不能有空格，否则会警告：WARNING:root:could not open file '/etc/apt/sources.list.d，并报错

原网站

版权声明
本文为[是七叔呀]所创，转载请带上原文链接，感谢
https://blog.csdn.net/m0_46378271/article/details/125178470

边栏推荐

猜你喜欢

随机推荐