Skip to content

关于使用GPU测试代码遇到的一些问题 #5

@Pandoravirus-N-T

Description

@Pandoravirus-N-T

尊敬的作者你好,我在尝试复现测试代码时遇见了一些问题。
我尝试使用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运行,这似乎很奇怪。
希望得到您的回复

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions