首先,kaggle已经为我们搭建好了cuda平台,但如果我们直接下载cuml便会报错。再接着你想去看看你的cuda是否存在时,会发现不存在。
解决办法:
首先要先创建一个新的notebook
第一步:
一定不要把kaggle提供的gpu打开:
将GPU改为none!!!
第二步:
下载cuml:
直接下载通常会报错,我们首先打开Installation Guide - RAPIDS Docs该网页
如图
再接着往下滑找到如图:
选择你对应的需要
提示!!!最好能不下载standard版本就不下载,因为下载后会有很多依赖包冲突,一般来说使用cuml提供的svm方法下载和我的一样的就可以 记得在这个命令前加 !
第三步:
import torch
print(torch.version.cuda)
输入这个
你会看到他会输出“none”
然后我们把gpu打开
这个时候我们再把我们自己的代码运行就可以了!!!
一定要按照步骤来