顺德外贸网站建设教育机构加盟

安装了GPU版本的pytorch的时候，想要使用CPU，怎么操作呢？
设置环境变量：

set TF_FORCE_GPU_ALLOW_GROWTH=false
set CUDA_VISIBLE_DEVICES=''

如果想要使用固定序号的GUP设备，则指定ID

set CUDA_VISIBLE_DEVICES='0'  # 使用第0块GPU

注意: 如果是window系统，则是 set，如果是基于linux系统，则是 export