因为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])三、
下面是在租的
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)