如果安装的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
您可能感兴趣的与本文相关的镜像
Python3.11
Python 是一种高级、解释型、通用的编程语言,以其简洁易读的语法而闻名,适用于广泛的应用,包括Web开发、数据分析、人工智能和自动化脚本
4586
4万+
6014
484
2万+

被折叠的 条评论
为什么被折叠?