pytorch安装GPU版本,指定设备
安装了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