尊敬的作者你好,我在尝试复现测试代码时遇见了一些问题。
我尝试使用CUDA_VISIBLE_DEVICES=0 python test.py --input_folder ./test_data --same_folder ./output_QuadPrior命令时发现不能在windows系统应用。但最令我不解地是我直接运行test.py文件时使用内存发现不够,而gpu内存也未使用,能问问大概需要多少内存。或者如何修改代码让测试代码在gpu内存上运行,因为gpu运行内存都足够推理约13GB的使用。同时我在test文件添加
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
这段代码运行时发现在调用cpu运行,这似乎很奇怪。
希望得到您的回复
尊敬的作者你好,我在尝试复现测试代码时遇见了一些问题。
我尝试使用CUDA_VISIBLE_DEVICES=0 python test.py --input_folder ./test_data --same_folder ./output_QuadPrior命令时发现不能在windows系统应用。但最令我不解地是我直接运行test.py文件时使用内存发现不够,而gpu内存也未使用,能问问大概需要多少内存。或者如何修改代码让测试代码在gpu内存上运行,因为gpu运行内存都足够推理约13GB的使用。同时我在test文件添加
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
这段代码运行时发现在调用cpu运行,这似乎很奇怪。
希望得到您的回复