当我保存训练模型时，cuML随机森林分割故障(核丢弃)

Question

我试图保存我的射频模型后，它的训练，我得到了一个“分割错误(核心抛出)”。我试着在训练前把它保存下来，这并不会给我带来任何问题。我尝试过其他的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 ->分段错误(核心转储)

系统规格：

1. Ubuntu 18.04
2. RAM 32 RAM
3. NVIDIA-SMI 495.29.05驱动程序版本: 495.29.05 CUDA版本: 11.5Cuda信息1：https://i.stack.imgur.com/8reIM.png
4. 急流22.04
5. Python 3.9
6. CPU: Intel Core i7-7700
7. GPU: NVIDIA GeForce RTX 2080超级8GB
8. 酸菜4.0

Answer 1

解决了

经过一些研究，我更新了急流版本为22.06，现在允许 me保存经过训练的带有泡菜的随机森林模型。