tensorflow/pytorch/指定GPU訓練的幾種方式

tensorflow/pytorch/指定GPU訓練的幾種方式 一、python代碼中添加:python os.environ["CUDA_VISIBLE_DEVICES"] = "0" orweb # CUDA_VISIBLE_DEVICES=1 二、 命令窗口臨時限制環境變量bash export CUDA_VISIBLE_DEVICES=1 要想應用到全部用戶,須要在~/.bashrc中最後
相關文章
相關標籤/搜索