登录社区云,与社区用户共同成长
邀请您加入社区
笔者电脑系统ubuntu18.04,cuda环境10.2,cudnn7.6.5,环境配置之后clion新建的cuda工程cmake报错:No CMAKE_CUDA_COMPILER could be found。原因是因为没有找到cuda的编译器,解决很简单,在clion中手动配置cuda路径,详细如图:在file->setting->build->cmake,在cmake op
3、高版本pytorch可能在处理显存占用时有更多的优化,可以升级为更高版本的pytorch。特别是在相同型号GPU的情况下,可以考虑Python包版本的问题,通过“pip list”或者“conda list”可以参考版本信息。对于1,在尝试减少batchsize解决,发现原文中原本的batchsize已经是1,没有办法再减了。但是此方法只是应急用,至于怎么让他多个GPU一起用,我至今还没有弄明
是在这一步遇到的:di = net(batch["image"].to(device))这两个问题我调试了1天半:结果发现!都是cuda的问题,# 确保所有输入都在指定设备上。