- 博客(2)
- 收藏
- 关注
原创 NVIDIA GeForce RTX 3050 with CUDA capability sm_86 is not compatible with the current PyTorch instal
2.卸载之前的torch和torchvision,pip uninstall torch/torchvison;3:重新下载1.8.0的torch和0.9.0的torchvision(建议下载whl,如上链接);4.进入虚拟环境,进入到whl存放的路径,我的是 D:->cd app-pc->cd conda。查阅博客了解到30系不支持直接使用CUDA10,我的电脑cuda版本为10.2。1.卸载cuda,去官网下载11.1版本的cuda(百度);6.进入虚拟环境测试成功。
2023-10-01 15:30:37 642
原创 pytorch根据霹雳吧啦复现AlexNet报错__init__ got an excepted keyword argument ‘num_classes
model中需要传入分类个数的参数,在train中一直提示出现出现意外参数,并报错__init__ got an excepted keyword argument 'num_classes。原因:model中def __init__和super.__init__写成了__int__
2023-10-01 13:39:38 327
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人