问题

pytorch在项目里 pip install clip之后 import clip报错（infer时）：
“Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you must use the ‘spawn’ start method”

解决

查阅位置跟multi process有关

for i in range(thread_num):p = mp.Process(target=sub_processor, args=(lock, i, args, data, save_path_prefix, save_visualize_path_prefix, img_folder, sub_video_list))

网上解决办法：

torch.multiprocessing.set_start_method(‘spawn’)

修改之后报错：

TypeError: cannot pickle ‘_thread.lock’ object

==============================================================================
逐行排查代码后发现与上述mp无关，把CLIP的load函数的参数从：

def load(name: str, device: Union[str, torch.device] = “cuda” if torch.cuda.is_available() else “cpu”, …

修改为（直接默认为cuda了因为我暂时没有cpu的需求）

def load(name: str, device: Union[str, torch.device] = “cuda”, …

import clip时Cannot re-initialize CUDA in forked subprocess相关推荐

Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you
RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you ...
cannot re-initialize CUDA in forked subprocess.To use CUDA with multiprocessing,you must use the ...
cannot re-initialize CUDA in forked subprocess.To use CUDA with multiprocessing,you must use the 'sp ...
RuntimeError: Cannot re-initialize CUDA in forked subprocess
问题就在于训练所使用的num_workers数量问题,直接将 num_workers=0 就不会报错了
Pytorch 训练与测试时爆显存(cuda out of memory)的终极解决方案，使用cpu(勿喷)
Pytorch 训练与测试时爆显存(cuda out of memory)的终极解决方案,使用cpu(勿喷) 参见了很多方法,都没有用. 简单点,直接把gpu设成-1
在使用import语句时
在使用import语句时,总出现问题,就上网找了一下,在csdn上一个叫Stillbright发的,8错,分享一下!(你也可以参考http://tb.blog.csdn.net/TrackBack.a ...
【Python】import pandas时，报错 pandas Missing required dependencies ['numpy'] 原因分析
结论:在需要 import pandas 时,同级目录下不要有 unittest.py同名文件,否则,会与pandas自带的unittest冲突,导致 import pandas 失败. 错误使用场景 ...
ImportError: DLL load failed: 找不到指定的模块。Import tensorflow时
在Import tensorflow时出现这个错误需要下载visual c++ 链接: [https://support.microsoft.com/en-us/help/2977003/the-l ...
import cv2时出现ImportError: DLL load fail:找不到指定模块
一.问题描述: 运行代码提示"import cv2时出现ImportError: DLL load fail:找不到指定模块". 原因分析:找不到某个注册表. 二.解决方法: 1. ...
import xlwings时_pickle.UnpicklingError: invalid load key, ‘\x00‘解决方案
大家好,我是herosunly.985院校硕士毕业,现担任算法研究员一职,热衷于机器学习算法研究与应用.曾获得阿里云天池比赛第一名,科大讯飞比赛第三名,CCF比赛第四名.拥有多项发明专利.对机器学 ...
from import 导入时找不到module的解决办法（Python模块包中_init_.py文件的作用）
文章目录背景正文 Ⅰ.报错信息 Ⅱ.问题解决 Ⅲ.问题分析 Ⅳ._init_.py文件作用背景最近在使用Pycharm开发python脚本时,遇到这样一个现象:打开别人给的工程,环境配置都没有 ...

import clip时Cannot re-initialize CUDA in forked subprocess

问题

解决

import clip时Cannot re-initialize CUDA in forked subprocess相关推荐

最新文章

热门文章