无法将cudf、cupy和cuml安装到rapids.ai版本为21.08的colab中

Question

我正在尝试按照这个教程在google colab pro上安装cudf和cuml：rapids_cudf.ipynb - Colaboratory

但在运行以下代码块之后：

# intall miniconda
!wget -c https://repo.continuum.io/miniconda/Miniconda3-4.5.4-Linux-x86_64.sh
!chmod +x Miniconda3-4.5.4-Linux-x86_64.sh
!bash ./Miniconda3-4.5.4-Linux-x86_64.sh -b -f -p /usr/local

# install RAPIDS packages
!conda install -q -y --prefix /usr/local -c conda-forge \
  -c rapidsai-nightly/label/cuda10.0 -c nvidia/label/cuda10.0 \
  cudf cuml

# set environment vars
import sys, os, shutil
sys.path.append('/usr/local/lib/python3.6/site-packages/')
os.environ['NUMBAPRO_NVVM'] = '/usr/local/cuda/nvvm/lib64/libnvvm.so'
os.environ['NUMBAPRO_LIBDEVICE'] = '/usr/local/cuda/nvvm/libdevice/'

# copy .so files to current working dir
for fn in ['libcudf.so', 'librmm.so']:
  shutil.copy('/usr/local/lib/'+fn, os.getcwd())

我收到以下错误：

---------------------------------------------------------------------------
FileNotFoundError                         Traceback (most recent call last)
<ipython-input-3-f6f0b40601c8> in <module>()
     15 # copy .so files to current working dir
     16 for fn in ['libcudf.so', 'librmm.so']:
---> 17   shutil.copy('/usr/local/lib/'+fn, os.getcwd())

1 frames
/usr/lib/python3.7/shutil.py in copyfile(src, dst, follow_symlinks)
    118         os.symlink(os.readlink(src), dst)
    119     else:
--> 120         with open(src, 'rb') as fsrc:
    121             with open(dst, 'wb') as fdst:
    122                 copyfileobj(fsrc, fdst)

FileNotFoundError: [Errno 2] No such file or directory: '/usr/local/lib/libcudf.so'

我有以下GPU: B‘’Tesla P100-PCIE-16 on‘on colab pro。

在运行上面给出的代码块之后，python版本是：Python 3.6.5 :: Anaconda, Inc.

以下是其他规范：

# Check Ubuntu Version
!lsb_release -a

输出：

No LSB modules are available.
Distributor ID: Ubuntu
Description:    Ubuntu 18.04.5 LTS
Release:    18.04
Codename:   bionic

# Check CUDA/cuDNN Version
!nvcc -V && which nvcc

输出：

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Oct_12_20:09:46_PDT_2020
Cuda compilation tools, release 11.1, V11.1.105
Build cuda_11.1.TC455_06.29190527_0
/usr/local/cuda/bin/nvcc

我已经尝试遵循StackOverflow的这个教程，但无法安装cudf或cuml。

Answer 1

此文件正在使用过时的安装代码和不兼容的CUDA选项。我一定会让维护人员更新它的。请使用rapids-colab template中最新的代码：https://colab.research.google.com/drive/1TAAi_szMfWqRfHVfjGSqnGVLr_ztzUM9#scrollTo=G_2dancFhf70。