比如说,程序使用第 2 张显卡(从 0 开始计数)。它的作用是告诉系统和深度学习框架(如 PyTorch 或 TensorFlow)只可见某些 GPU。
export CUDA_VISIBLE_DEVICES=1
然后再查看当前使用的显卡:
echo $CUDA_VISIBLE_DEVICES
此时应该输出1:
通过以下命令查看使用显卡的状态:
nvitop
比如说,程序使用第 2 张显卡(从 0 开始计数)。它的作用是告诉系统和深度学习框架(如 PyTorch 或 TensorFlow)只可见某些 GPU。
export CUDA_VISIBLE_DEVICES=1
然后再查看当前使用的显卡:
echo $CUDA_VISIBLE_DEVICES
此时应该输出1:
通过以下命令查看使用显卡的状态:
nvitop
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:/a/936058.html
如若内容造成侵权/违法违规/事实不符,请联系我们进行投诉反馈qq邮箱809451989@qq.com,一经查实,立即删除!