当我保存经过训练的模型时，cuML Random Forest- Segmentation fault（core dumped）

Question

我试图在训练后保存我的 Rf 模型，但出现“分段错误（核心转储）”。 我曾尝试在训练前保存它，它不会给我带来任何泡菜问题。 我尝试过其他cuML算法，它让我在训练后保存它们。

from cuml.ensemble  import  RandomForestClassifier as rf_cuML
import pickle

model = rf_cuML(random_state=0)
output = open('model.pkl', 'wb')
pickle.dump(model, output) #This saving works

model.fit(train_features, train_labels)
output = open('model_trained.pkl', 'wb')
pickle.dump(model, output) #This saving does not work

此行抛出以下错误： pickle.dump(model, output) #This saving does not work --> Segmentation fault (core dumped)

系统规格：

1. Ubuntu 18.04
2. 内存 32GB
3. NVIDIA-SMI 495.29.05 驱动程序版本：495.29.05 CUDA 版本：11.5[Cuda info][1] [1]：https://i.stack.imgur.com/8reIM.png
4. 急流 22.04
5. 蟒蛇 3.9
6. CPU：英特尔酷睿 i7-7700
7. GPU：NVIDIA GeForce RTX 2080 超级 8 GB
8. 泡菜4.0

Answer 1

解决了

经过一番研究，我将rapids 版本更新为22.06 ，现在它允许我用pickle保存经过训练的随机森林模型。