美文网首页
cuda指定某块gpu运行程序

cuda指定某块gpu运行程序

作者: 胜负55开 | 来源:发表于2019-10-28 19:48 被阅读0次

    (1)先查看当前服务器下的显卡情况:nvidia-smi

    图1:显卡情况

    (2)用如下命令指定某张卡,来运行程序:
    CUDA_VISIBLE_DEVICES=4 python main.py --cuda

    相关文章

      网友评论

          本文标题:cuda指定某块gpu运行程序

          本文链接:https://www.haomeiwen.com/subject/twfivctx.html