在 Kaggle 上配置 CUDA 的步骤如下:

  1. 在 Kaggle 上创建一个 Notebook。

  2. 在 Notebook 中选择 'Settings' 选项卡。

  3. 在 'Accelerator' 下拉菜单中选择 'GPU'。

  4. 点击 'Save' 按钮。

  5. 在 Notebook 中运行以下代码,以确保 CUDA 已安装并运行:

!nvidia-smi
  1. 如果 CUDA 未安装,则需要安装 CUDA。可以通过以下步骤进行安装:
!wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
!sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
!wget http://developer.download.nvidia.com/compute/cuda/11.4.1/local_installers/cuda-repo-ubuntu1804-11-4-local_11.4.1-470.57.02-1_amd64.deb
!sudo dpkg -i cuda-repo-ubuntu1804-11-4-local_11.4.1-470.57.02-1_amd64.deb
!sudo apt-key add /var/cuda-repo-ubuntu1804-11-4-local/7fa2af80.pub
!sudo apt-get update
!sudo apt-get -y install cuda
  1. 安装完成后,重新运行步骤 5 的代码,以确保 CUDA 已正确安装并运行。

注意:在 Kaggle 上使用 CUDA 可能需要一些时间来完成,因此请耐心等待。


原文地址: https://www.cveoy.top/t/topic/oHzF 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录