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中最后
相关文章
相关标签/搜索