如果安装的Pytorch是cpu版本的,不是gpu版本,就在python命令后面加上 --gpu_ids -1(gpu_ids原来默认值是0),问题解决;
例如:
python train.py --data_dir sample_dataset --gpu_ids -1
如果安装的Pytorch是cpu版本的,不是gpu版本,就在python命令后面加上 --gpu_ids -1(gpu_ids原来默认值是0),问题解决;
例如:
python train.py --data_dir sample_dataset --gpu_ids -1