深度学习学习笔记——在kaggle上跑mmdetection

深度学习学习笔记——在kaggle上跑mmdetection,第1张

深度学习学习笔记——在kaggle上跑mmdetection 一、pytorch版本升级:

因为kaggle默认的torch版本时1.4,而mmcv需要的最低torch版本大于1.4,故要升级torch版本,但是刚点进pytorch官网发现支持的cuda版本只有10.2和11.3,所以只能另想办法,最后用pip install torch==1.7的方式指定torch版本,最后虽然下载下来了发现用pip show pytorch确实是下载的1.7版本,可是用torch.__version__发现还是原来的1.4版本,没弄明白但当然不能用,最后发现官网原来是有其他指定的cuda版本对应的指定torch版本下载的,老老实实使用官网指定方法下载后成功了
官网链接:https://pytorch.org/get-started/previous-versions/

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
二、kernel

kaggle目录: /——kaggle——input、lib、working

kaggle卸载包: kaggle中的notebook卸载包时没法交互也就没法yes,所以可以按如下方法解决:

import pip

def pip_install(package):
   pip.main(['install', package])
def pip_list():
   pip.main(['list'])
def pip_uninstall(package):
    pip.main(['uninstall', package])
三、




下面是在租的

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5670824.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-16
下一篇 2022-12-16

发表评论

登录后才能评论

评论列表(0条)

保存