cuML包在jupyterlab中导入耗时较长
我正在尝试将以下功能导入到我的jupyterlab工作区：from cuml.manifold import TSNE
有时候这个功能加载得很顺利，但有时候却要花很长时间（或者根本不动 - 我每次尝试大约等了5分钟就停止了）。我不太明白这里出了什么问题。
为了让你们更了解情况，我想说明一下，因为这是一个只适用于Linux的包，所以我在我的Windows机器上通过WSL2的conda环境来使用它。我的代码中的其他部分运行得很好，但我就是不明白为什么这一部分不行？
我安装了CUDA 12.4，驱动版本是551.61。
我使用conda安装了这个包，命令是使用夜间发布版本：
conda create --solver=libmamba -n rapids-24.04 -c rapidsai-nightly -c conda-forge -c nvidia \
rapids=24.04 python=3.10 cuda-version=12.2 \
jupyterlab dash pytorch
环境中安装的包：
# Name Version Build Channel
_libgcc_mutex 0.1 conda_forge conda-forge
_openmp_mutex 4.5 2_kmp_llvm conda-forge
anyio 4.3.0 pyhd8ed1ab_0 conda-forge
argon2-cffi 23.1.0 pyhd8ed1ab_0 conda-forge
argon2-cffi-bindings 21.2.0 py310h2372a71_4 conda-forge
arrow 1.3.0 pyhd8ed1ab_0 conda-forge
asttokens 2.4.1 pyhd8ed1ab_0 conda-forge
async-lru 2.0.4 pyhd8ed1ab_0 conda-forge
attrs 23.2.0 pyh71513ae_0 conda-forge
aws-c-auth 0.7.16 h79b3bcb_6 conda-forge
aws-c-cal 0.6.10 hb29e0c7_1 conda-forge
aws-c-common 0.9.13 hd590300_0 conda-forge
aws-c-compression 0.2.18 hecc5fa9_1 conda-forge
aws-c-event-stream 0.4.2 hf9b2f7b_4 conda-forge
aws-c-http 0.8.1 h5d7533a_5 conda-forge
aws-c-io 0.14.5 h50678d4_1 conda-forge
aws-c-mqtt 0.10.2 hf479d2b_4 conda-forge
aws-c-s3 0.5.2 h4ad9680_0 conda-forge
aws-c-sdkutils 0.1.15 hecc5fa9_1 conda-forge
aws-checksums 0.1.18 hecc5fa9_1 conda-forge
aws-crt-cpp 0.26.2 h19f5d62_7 conda-forge
aws-sdk-cpp 1.11.267 h5606698_1 conda-forge
babel 2.14.0 pyhd8ed1ab_0 conda-forge
beautifulsoup4 4.12.3 pyha770c72_0 conda-forge
bleach 6.1.0 pyhd8ed1ab_0 conda-forge
blinker 1.7.0 pyhd8ed1ab_0 conda-forge
bokeh 3.3.4 pyhd8ed1ab_0 conda-forge
brotli-python 1.1.0 py310hc6cd4ac_1 conda-forge
bzip2 1.0.8 hd590300_5 conda-forge
c-ares 1.27.0 hd590300_0 conda-forge
ca-certificates 2024.2.2 hbcca054_0 conda-forge
cached-property 1.5.2 hd8ed1ab_1 conda-forge
cached_property 1.5.2 pyha770c72_1 conda-forge
cachetools 5.3.3 pyhd8ed1ab_0 conda-forge
certifi 2024.2.2 py310h06a4308_0
cffi 1.16.0 py310h2fee648_0 conda-forge
charset-normalizer 3.3.2 pyhd8ed1ab_0 conda-forge
click 8.1.7 unix_pyh707e725_0 conda-forge
cloudpickle 3.0.0 pyhd8ed1ab_0 conda-forge
comm 0.2.1 pyhd8ed1ab_0 conda-forge
condacolab 0.1.9 pypi_0 pypi
contourpy 1.2.0 py310hd41b1e2_0 conda-forge
cuda-cccl_linux-64 12.2.140 ha770c72_0 conda-forge
cuda-crt-dev_linux-64 12.2.140 ha770c72_1 conda-forge
cuda-crt-tools 12.2.140 ha770c72_1 conda-forge
cuda-cudart 12.2.140 hd3aeb46_0 conda-forge
cuda-cudart-dev 12.2.140 hd3aeb46_0 conda-forge
cuda-cudart-dev_linux-64 12.2.140 h59595ed_0 conda-forge
cuda-cudart-static 12.2.140 hd3aeb46_0 conda-forge
cuda-cudart-static_linux-64 12.2.140 h59595ed_0 conda-forge
cuda-cudart_linux-64 12.2.140 h59595ed_0 conda-forge
cuda-nvcc-dev_linux-64 12.2.140 ha770c72_1 conda-forge
cuda-nvcc-impl 12.2.140 hd3aeb46_1 conda-forge
cuda-nvcc-tools 12.2.140 hd3aeb46_1 conda-forge
cuda-nvrtc 12.2.140 hd3aeb46_0 conda-forge
cuda-nvtx 12.2.140 h59595ed_0 conda-forge
cuda-nvvm-dev_linux-64 12.2.140 ha770c72_1 conda-forge
cuda-nvvm-impl 12.2.140 h59595ed_1 conda-forge
cuda-nvvm-tools 12.2.140 h59595ed_1 conda-forge
cuda-profiler-api 12.2.140 ha770c72_0 conda-forge
cuda-python 12.3.0 py310h9f9f131_2 conda-forge
cuda-version 12.2 he2b69de_2 conda-forge
cudf 24.04.00a489 cuda12_py310_240304_ge60aad110e_489 rapidsai-nightly
cudnn 8.8.0.121 h264754d_4 conda-forge
cuml 24.04.00a34 cuda12_py310_240304_gb0ba34091_34 rapidsai-nightly
cupy 13.0.0 py310h7aad9d2_3 conda-forge
cupy-core 13.0.0 py310had4011e_3 conda-forge
cytoolz 0.12.3 py310h2372a71_0 conda-forge
dash 2.16.0 pyhd8ed1ab_0 conda-forge
dask 2024.1.1 pyhd8ed1ab_0 conda-forge
dask-core 2024.1.1 pyhd8ed1ab_0 conda-forge
dask-cuda 24.04.00a11 py310_240304_g6be9df9_11 rapidsai-nightly
dask-cudf 24.04.00a489 cuda12_py310_240304_ge60aad110e_489 rapidsai-nightly
dask-expr 0.4.0 pyhd8ed1ab_0 conda-forge
debugpy 1.8.1 py310hc6cd4ac_0 conda-forge
decorator 5.1.1 pyhd8ed1ab_0 conda-forge
defusedxml 0.7.1 pyhd8ed1ab_0 conda-forge
distributed 2024.1.1 pyhd8ed1ab_0 conda-forge
dlpack 0.5 h9c3ff4c_0 conda-forge
entrypoints 0.4 pyhd8ed1ab_0 conda-forge
exceptiongroup 1.2.0 pyhd8ed1ab_2 conda-forge
executing 2.0.1 pyhd8ed1ab_0 conda-forge
fastrlock 0.8.2 py310hc6cd4ac_2 conda-forge
filelock 3.13.1 pyhd8ed1ab_0 conda-forge
flask 3.0.2 pyhd8ed1ab_0 conda-forge
fmt 10.2.1 h00ab1b0_0 conda-forge
fqdn 1.5.1 pyhd8ed1ab_0 conda-forge
freetype 2.12.1 h267a509_2 conda-forge
fsspec 2024.2.0 pyhca7485f_0 conda-forge
gflags 2.2.2 he1b5a44_1004 conda-forge
glog 0.7.0 hed5481d_0 conda-forge
gmock 1.14.0 ha770c72_1 conda-forge
gmp 6.3.0 h59595ed_0 conda-forge
gmpy2 2.1.2 py310h3ec546c_1 conda-forge
gtest 1.14.0 h00ab1b0_1 conda-forge
h11 0.14.0 pyhd8ed1ab_0 conda-forge
h2 4.1.0 pyhd8ed1ab_0 conda-forge
hpack 4.0.0 pyh9f0ad1d_0 conda-forge
httpcore 1.0.4 pyhd8ed1ab_0 conda-forge
httpx 0.27.0 pyhd8ed1ab_0 conda-forge
hyperframe 6.0.1 pyhd8ed1ab_0 conda-forge
icu 73.2 h59595ed_0 conda-forge
idna 3.6 pyhd8ed1ab_0 conda-forge
importlib-metadata 7.0.1 pyha770c72_0 conda-forge
importlib_metadata 7.0.1 hd8ed1ab_0 conda-forge
importlib_resources 6.1.2 pyhd8ed1ab_0 conda-forge
ipykernel 6.29.3 pyhd33586a_0 conda-forge
ipython 8.22.1 pyh707e725_0 conda-forge
isoduration 20.11.0 pyhd8ed1ab_0 conda-forge
itsdangerous 2.1.2 pyhd8ed1ab_0 conda-forge
jedi 0.19.1 pyhd8ed1ab_0 conda-forge
jinja2 3.1.3 pyhd8ed1ab_0 conda-forge
joblib 1.3.2 pyhd8ed1ab_0 conda-forge
json5 0.9.20 pyhd8ed1ab_0 conda-forge
jsonpointer 2.4 py310hff52083_3 conda-forge
jsonschema 4.21.1 pyhd8ed1ab_0 conda-forge
jsonschema-specifications 2023.12.1 pyhd8ed1ab_0 conda-forge
jsonschema-with-format-nongpl 4.21.1 pyhd8ed1ab_0 conda-forge
jupyter-lsp 2.2.3 pyhd8ed1ab_0 conda-forge
jupyter_client 8.6.0 pyhd8ed1ab_0 conda-forge
jupyter_core 5.7.1 py310hff52083_0 conda-forge
jupyter_events 0.9.0 pyhd8ed1ab_0 conda-forge
jupyter_server 2.12.5 pyhd8ed1ab_0 conda-forge
jupyter_server_terminals 0.5.2 pyhd8ed1ab_0 conda-forge
jupyterlab 4.1.2 pyhd8ed1ab_0 conda-forge
jupyterlab_pygments 0.3.0 pyhd8ed1ab_1 conda-forge
jupyterlab_server 2.25.3 pyhd8ed1ab_0 conda-forge
keyutils 1.6.1 h166bdaf_0 conda-forge
krb5 1.21.2 h659d440_0 conda-forge
lcms2 2.16 hb7c19ff_0 conda-forge
ld_impl_linux-64 2.40 h41732ed_0 conda-forge
lerc 4.0.0 h27087fc_0 conda-forge
libabseil 20230802.1 cxx17_h59595ed_0 conda-forge
libarrow 14.0.2 h17fe1ab_9_cpu conda-forge
libarrow-acero 14.0.2 h59595ed_9_cpu conda-forge
libarrow-dataset 14.0.2 h59595ed_9_cpu conda-forge
libarrow-flight 14.0.2 hdc44a87_9_cpu conda-forge
libarrow-flight-sql 14.0.2 hfbc7f12_9_cpu conda-forge
libarrow-gandiva 14.0.2 h308e607_9_cpu conda-forge
libarrow-substrait 14.0.2 hfbc7f12_9_cpu conda-forge
libblas 3.9.0 21_linux64_openblas conda-forge
libbrotlicommon 1.1.0 hd590300_1 conda-forge
libbrotlidec 1.1.0 hd590300_1 conda-forge
libbrotlienc 1.1.0 hd590300_1 conda-forge
libcblas 3.9.0 21_linux64_openblas conda-forge
libcrc32c 1.1.2 h9c3ff4c_0 conda-forge
libcublas 12.2.5.6 hd3aeb46_0 conda-forge
libcublas-dev 12.2.5.6 hd3aeb46_0 conda-forge
libcudf 24.04.00a489 cuda12_240304_ge60aad110e_489 rapidsai-nightly
libcufft 11.0.8.103 hd3aeb46_0 conda-forge
libcufile 1.7.2.10 hd3aeb46_0 conda-forge
libcufile-dev 1.7.2.10 hd3aeb46_0 conda-forge
libcuml 24.04.00a34 cuda12_240304_gb0ba34091_34 rapidsai-nightly
libcumlprims 24.04.00a cuda12_240304_gc4a7a87_9 rapidsai-nightly
libcurand 10.3.3.141 hd3aeb46_0 conda-forge
libcurand-dev 10.3.3.141 hd3aeb46_0 conda-forge
libcurl 8.5.0 hca28451_0 conda-forge
libcusolver 11.5.2.141 hd3aeb46_0 conda-forge
libcusolver-dev 11.5.2.141 hd3aeb46_0 conda-forge
libcusparse 12.1.2.141 hd3aeb46_0 conda-forge
libcusparse-dev 12.1.2.141 hd3aeb46_0 conda-forge
libdeflate 1.19 hd590300_0 conda-forge
libedit 3.1.20191231 he28a2e2_2 conda-forge
libev 4.33 hd590300_2 conda-forge
libevent 2.1.12 hf998b51_1 conda-forge
libffi 3.4.2 h7f98852_5 conda-forge
libgcc-ng 13.2.0 h807b86a_5 conda-forge
libgfortran-ng 13.2.0 h69a702a_5 conda-forge
libgfortran5 13.2.0 ha4646dd_5 conda-forge
libgoogle-cloud 2.17.0 h31df0ca_2 conda-forge
libgoogle-cloud-storage 2.17.0 hc7a4891_2 conda-forge
libgrpc 1.60.1 h74775cd_0 conda-forge
libhwloc 2.9.3 default_h554bfaf_1009 conda-forge
libiconv 1.17 hd590300_2 conda-forge
libjpeg-turbo 3.0.0 hd590300_1 conda-forge
libkvikio 24.04.00a cuda12_240304_g787a6a3_18 rapidsai-nightly
liblapack 3.9.0 21_linux64_openblas conda-forge
libllvm14 14.0.6 hcd5def8_4 conda-forge
libllvm15 15.0.7 hb3ce162_4 conda-forge
libmagma 2.7.2 h173bb3b_2 conda-forge
libmagma_sparse 2.7.2 h173bb3b_2 conda-forge
libnghttp2 1.58.0 h47da74e_1 conda-forge
libnl 3.9.0 hd590300_0 conda-forge
libnsl 2.0.1 hd590300_0 conda-forge
libnuma 2.0.16 h0b41bf4_1 conda-forge
libnvjitlink 12.2.140 hd3aeb46_0 conda-forge
libopenblas 0.3.26 pthreads_h413a1c8_0 conda-forge
libparquet 14.0.2 h352af49_9_cpu conda-forge
libpng 1.6.43 h2797004_0 conda-forge
libprotobuf 4.25.1 hf27288f_2 conda-forge
libraft 24.04.00a66 cuda12_240304_gad27a731_66 rapidsai-nightly
libraft-headers 24.04.00a66 cuda12_240304_gad27a731_66 rapidsai-nightly
libraft-headers-only 24.04.00a66 cuda12_240304_gad27a731_66 rapidsai-nightly
libre2-11 2023.09.01 h7a70373_1 conda-forge
librmm 24.04.00a32 cuda12_240304_gf132d4b0_32 rapidsai-nightly
libsodium 1.0.18 h36c2ea0_1 conda-forge
libsqlite 3.45.1 h2797004_0 conda-forge
libssh2 1.11.0 h0841786_0 conda-forge
libstdcxx-ng 13.2.0 h7e041cc_5 conda-forge
libthrift 0.19.0 hb90f79a_1 conda-forge
libtiff 4.6.0 ha9c0a0a_2 conda-forge
libtorch 2.1.2 cuda120_h2aa5df7_301 conda-forge
libutf8proc 2.8.0 h166bdaf_0 conda-forge
libuuid 2.38.1 h0b41bf4_0 conda-forge
libuv 1.46.0 hd590300_0 conda-forge
libwebp-base 1.3.2 hd590300_0 conda-forge
libxcb 1.15 h0b41bf4_0 conda-forge
libxcrypt 4.4.36 hd590300_1 conda-forge
libxml2 2.12.5 h232c23b_0 conda-forge
libzlib 1.2.13 hd590300_5 conda-forge
llvm-openmp 17.0.6 h4dfa4b3_0 conda-forge
llvmlite 0.42.0 py310h1b8f574_1 conda-forge
locket 1.0.0 pyhd8ed1ab_0 conda-forge
lz4 4.3.3 py310h350c4a5_0 conda-forge
lz4-c 1.9.4 hcb278e6_0 conda-forge
magma 2.7.2 h51420fd_2 conda-forge
markdown-it-py 3.0.0 pyhd8ed1ab_0 conda-forge
markupsafe 2.1.5 py310h2372a71_0 conda-forge
matplotlib 3.8.3 pypi_0 pypi
matplotlib-inline 0.1.6 pyhd8ed1ab_0 conda-forge
mdurl 0.1.2 pyhd8ed1ab_0 conda-forge
mistune 3.0.2 pyhd8ed1ab_0 conda-forge
mkl 2023.2.0 h84fe81f_50496 conda-forge
mpc 1.3.1 hfe3b2da_0 conda-forge
mpfr 4.2.1 h9458935_0 conda-forge
mpmath 1.3.0 pyhd8ed1ab_0 conda-forge
msgpack-python 1.0.7 py310hd41b1e2_0 conda-forge
nbclient 0.8.0 pyhd8ed1ab_0 conda-forge
nbconvert-core 7.16.1 pyhd8ed1ab_0 conda-forge
nbformat 5.9.2 pyhd8ed1ab_0 conda-forge
nccl 2.20.3.1 h3a97aeb_0 conda-forge
ncurses 6.4 h59595ed_2 conda-forge
nest-asyncio 1.6.0 pyhd8ed1ab_0 conda-forge
networkx 3.2.1 pyhd8ed1ab_0 conda-forge
nodejs 20.9.0 hb753e55_0 conda-forge
notebook-shim 0.2.4 pyhd8ed1ab_0 conda-forge
numba 0.59.0 py310h7dc5dd1_1 conda-forge
numpy 1.26.4 py310hb13e2d6_0 conda-forge
nvcomp 3.0.6 h10b603f_0 conda-forge
nvtx 0.2.10 py310h2372a71_0 conda-forge
openjpeg 2.5.2 h488ebb8_0 conda-forge
openssl 3.2.1 hd590300_0 conda-forge
orc 1.9.2 h7829240_1 conda-forge
overrides 7.7.0 pyhd8ed1ab_0 conda-forge
packaging 23.2 pyhd8ed1ab_0 conda-forge
pandas 2.2.1 py310hcc13569_0 conda-forge
pandocfilters 1.5.0 pyhd8ed1ab_0 conda-forge
parso 0.8.3 pyhd8ed1ab_0 conda-forge
partd 1.4.1 pyhd8ed1ab_0 conda-forge
pexpect 4.9.0 pyhd8ed1ab_0 conda-forge
pickleshare 0.7.5 py_1003 conda-forge
pillow 10.2.0 py310h01dd4db_0 conda-forge
pip 24.0 pyhd8ed1ab_0 conda-forge
pkgutil-resolve-name 1.3.10 pyhd8ed1ab_1 conda-forge
platformdirs 4.2.0 pyhd8ed1ab_0 conda-forge
plotly 5.19.0 pyhd8ed1ab_0 conda-forge
prometheus_client 0.20.0 pyhd8ed1ab_0 conda-forge
prompt-toolkit 3.0.42 pyha770c72_0 conda-forge
protobuf 4.25.1 py310h620c231_0 conda-forge
psutil 5.9.8 py310h2372a71_0 conda-forge
pthread-stubs 0.4 h36c2ea0_1001 conda-forge
ptyprocess 0.7.0 pyhd3deb0d_0 conda-forge
pure_eval 0.2.2 pyhd8ed1ab_0 conda-forge
pyarrow 14.0.2 py310hf9e7431_9_cpu conda-forge
pyarrow-hotfix 0.6 pyhd8ed1ab_0 conda-forge
pycparser 2.21 pyhd8ed1ab_0 conda-forge
pygments 2.17.2 pyhd8ed1ab_0 conda-forge
pylibraft 24.04.00a66 cuda12_py310_240304_gad27a731_66 rapidsai-nightly
pynvjitlink 0.1.13 py310hf694660_0 rapidsai-nightly
pynvml 11.4.1 pyhd8ed1ab_0 conda-forge
pyparsing 3.1.1 pypi_0 pypi
pysocks 1.7.1 pyha2e5f31_6 conda-forge
python 3.10.13 hd12c33a_1_cpython conda-forge
python-dateutil 2.9.0 pyhd8ed1ab_0 conda-forge
python-fastjsonschema 2.19.1 pyhd8ed1ab_0 conda-forge
python-json-logger 2.0.7 pyhd8ed1ab_0 conda-forge
python-tzdata 2024.1 pyhd8ed1ab_0 conda-forge
python_abi 3.10 4_cp310 conda-forge
pytorch 2.1.2 cuda120_py310h327d3bc_301 conda-forge
pytz 2024.1 pyhd8ed1ab_0 conda-forge
pyyaml 6.0.1 py310h2372a71_1 conda-forge
pyzmq 25.1.2 py310h795f18f_0 conda-forge
raft-dask 24.04.00a66 cuda12_py310_240304_gad27a731_66 rapidsai-nightly
rapids-dask-dependency 24.04.00a6 0 rapidsai-nightly
rdma-core 50.0 hd3aeb46_0 conda-forge
re2 2023.09.01 h7f4b329_1 conda-forge
readline 8.2 h8228510_1 conda-forge
referencing 0.33.0 pyhd8ed1ab_0 conda-forge
requests 2.31.0 pyhd8ed1ab_0 conda-forge
retrying 1.3.3 py_2 conda-forge
rfc3339-validator 0.1.4 pyhd8ed1ab_0 conda-forge
rfc3986-validator 0.1.1 pyh9f0ad1d_0 conda-forge
rich 13.7.1 pyhd8ed1ab_0 conda-forge
rmm 24.04.00a32 cuda12_py310_240304_gf132d4b0_32 rapidsai-nightly
rpds-py 0.18.0 py310hcb5633a_0 conda-forge
s2n 1.4.5 h06160fa_0 conda-forge
scipy 1.12.0 py310hb13e2d6_2 conda-forge
send2trash 1.8.2 pyh41d4057_0 conda-forge
setuptools 69.1.1 pyhd8ed1ab_0 conda-forge
six 1.16.0 pyh6c4a22f_0 conda-forge
sleef 3.5.1 h9b69904_2 conda-forge
snappy 1.1.10 h9fff704_0 conda-forge
sniffio 1.3.1 pyhd8ed1ab_0 conda-forge
sortedcontainers 2.4.0 pyhd8ed1ab_0 conda-forge
soupsieve 2.5 pyhd8ed1ab_1 conda-forge
spdlog 1.12.0 hd2e6256_2 conda-forge
stack_data 0.6.2 pyhd8ed1ab_0 conda-forge
sympy 1.12 pypyh9d50eac_103 conda-forge
tbb 2021.11.0 h00ab1b0_1 conda-forge
tblib 3.0.0 pyhd8ed1ab_0 conda-forge
tenacity 8.2.3 pyhd8ed1ab_0 conda-forge
terminado 0.18.0 pyh0d859eb_0 conda-forge
tinycss2 1.2.1 pyhd8ed1ab_0 conda-forge
tk 8.6.13 noxft_h4845f30_101 conda-forge
tomli 2.0.1 pyhd8ed1ab_0 conda-forge
toolz 0.12.1 pyhd8ed1ab_0 conda-forge
tornado 6.4 py310h2372a71_0 conda-forge
traitlets 5.14.1 pyhd8ed1ab_0 conda-forge
treelite 4.0.0 py310h4a6579d_0 conda-forge
types-python-dateutil 2.8.19.20240106 pyhd8ed1ab_0 conda-forge
typing-extensions 4.10.0 hd8ed1ab_0 conda-forge
typing_extensions 4.10.0 pyha770c72_0 conda-forge
typing_utils 0.1.0 pyhd8ed1ab_0 conda-forge
tzdata 2024a h0c530f3_0 conda-forge
ucx 1.15.0 h6d2d1ec_3 conda-forge
ucx-proc 1.0.0 gpu rapidsai-nightly
ucx-py 0.37.00a12 py310_240304_g388c7af_12 rapidsai-nightly
uri-template 1.3.0 pyhd8ed1ab_0 conda-forge
urllib3 2.2.1 pyhd8ed1ab_0 conda-forge
wcwidth 0.2.13 pyhd8ed1ab_0 conda-forge
webcolors 1.13 pyhd8ed1ab_0 conda-forge
webencodings 0.5.1 pyhd8ed1ab_2 conda-forge
websocket-client 1.7.0 pyhd8ed1ab_0 conda-forge
werkzeug 3.0.1 pyhd8ed1ab_0 conda-forge
wheel 0.42.0 pyhd8ed1ab_0 conda-forge
xorg-libxau 1.0.11 hd590300_0 conda-forge
xorg-libxdmcp 1.1.3 h7f98852_0 conda-forge
xyzservices 2023.10.1 pyhd8ed1ab_0 conda-forge
xz 5.2.6 h166bdaf_0 conda-forge
yaml 0.2.5 h7f98852_2 conda-forge
zeromq 4.3.5 h59595ed_1 conda-forge
zict 3.0.0 pyhd8ed1ab_0 conda-forge
zipp 3.17.0 pyhd8ed1ab_0 conda-forge
zlib 1.2.13 hd590300_5 conda-forge
zstd 1.5.5 hfc55251_0 conda-forge
1 个回答
0
我尝试重现这个问题，但没有成功。我创建了一个新的环境，使用了cuML的最新版本：
conda create --solver=libmamba -n rapids-24.04 -c rapidsai-nightly -c conda-forge -c nvidia \
cuml=24.04 python=3.10 cuda-version=12.0
然后我运行了 time python -c "import cuml" 100 次。第一次花了52秒，而之后的每次都不到5秒，所以我没有看到你提到的不稳定或变化。
一般来说，导入RAPIDS库可能比只用CPU的库要慢一点，因为需要花时间初始化CUDA和在GPU上加载代码。我不太清楚为什么第一次导入要花52秒，但我是在一个网络文件系统比较慢的系统上运行的。我猜大部分时间都是在从网络文件系统读取很多小文件，这些文件在后续调用时才被缓存。
如果你对这个问题有进一步的见解，或者找到重现这个明显卡住的方法，非常感谢你在这个问题上提供帮助：https://github.com/rapidsai/cuml/issues