求助，使用BVPP时出现问题！

tplover

错误代码：
【安装路径】\TG\basicvsrpp>basicvsrpp.exe --model_path models/st_80000.pth --input_path ../input0 --save_path ../output0 --interval 10 --device 1 --fraction 40
GPU_ID=1
Traceback (most recent call last):
  File "basicvsrpp.py", line 86, in <module>
  File "basicvsrpp.py", line 46, in main
  File "torch\cuda\memory.py", line 99, in set_per_process_memory_fraction
    torch._C._cuda_setMemoryFraction(fraction, device)
RuntimeError: 0 <= device && static_cast<size_t>(device) < device_allocator.size()INTERNAL ASSERT FAILED at "C:\\actions-runner\\_work\\pytorch\\pytorch\\builder\\windows\\pytorch\\c10\\cuda\\CUDACachingAllocator.cpp":1418, please report a bug to PyTorch. Allocator not initialized for device 1: did you call init?
[2368] Failed to execute script 'basicvsrpp' due to unhandled exception!

我的运行环境是：
RTX3050+CUDA 12.1.0.354+CUDNN8.9.0.131
在将使用的GPU从Inter核显调整为Nvidia独显之前它还能分成小块切割，但是那时候由于默认输出编码设置时GPU0：AMD，所以无法输出，但现在把GPU调整为GPU1 Nvidia之后甚至无法运行了。请大佬看看是CUDA安装问题吗？我看代码里好像调用的是CUDA10。

fredlin

请问你的问题搞好了吗？我也遇到这个。