广州网站seo营销模板,大数据平台网站建设,旅游网站建设与设计,重庆网站建设及优化公司比如说#xff0c;程序使用第 2 张显卡#xff08;从 0 开始计数#xff09;。它的作用是告诉系统和深度学习框架#xff08;如 PyTorch 或 TensorFlow#xff09;只可见某些 GPU。
export CUDA_VISIBLE_DEVICES1
然后再查看当前使用的显卡#xff1a;
echo $CUDA_VIS…比如说程序使用第 2 张显卡从 0 开始计数。它的作用是告诉系统和深度学习框架如 PyTorch 或 TensorFlow只可见某些 GPU。
export CUDA_VISIBLE_DEVICES1
然后再查看当前使用的显卡
echo $CUDA_VISIBLE_DEVICES此时应该输出1 通过以下命令查看使用显卡的状态
nvitop
