使用kaggle平台GPU运行自己本地项目
喵呜寨主
2023年12月25日 17:13

1.压缩本地代码文件,添加至个人数据集

cd /

ls -a

创建失败显示已存在目录,考虑修改数据集名称

2.安装第三方库(根据需要添加)

Kaggle服务器python版本是3.10,自带pytorch和相关的库。 如果想修改python版本,记得也下载对应的pytorch和相关的库

!conda create -n p310 python=3.10 -y

!sudo rm /opt/conda/bin/python

!sudo ln -s /opt/conda/envs/p310/bin/python3 /opt/conda/bin/python

!sudo rm /opt/conda/bin/pip

!sudo ln -s /opt/conda/envs/p310/bin/pip /opt/conda/bin/pip

建议还是使用Kaggle服务器自带的python,比较方便

第三方库可以通过github 或pip添加

git clone https://github.com/RUCAIBox/RecBole.git

cd RecBole

pip install -e . --verbose

3.移动代码至根目录,运行代码

cp -r /kaggle/input/rgcf-a3/RGCF-changed RGCF3

cd RGCF3

!python run_rgcf.py

如果运行速度很慢检查代码GPU设置

import torch

torch.cuda.get_device_name(0)