Ubuntu18.04 +cuda11+cudnn+opencv-3.1.0+3060/3090

3060/3090需要改变Nvidia驱动版本与cuda版本，升级到匹配版本，如果版本不匹配，即使编译成功会导致训练不收敛（亲测）。配置环境过程同https://blog.csdn.net/jdjjdjd/article/details/117731541碰到的问题为：编译opencv过程中报错，ippicv的包下载报错。由于翻墙的原因导致包下载出错，到opencv/3rdparty/ippicv

怀揣一梦

449人浏览 · 2021-06-09 10:13:28

怀揣一梦 · 2021-06-09 10:13:28 发布

3060/3090需要改变Nvidia驱动版本与cuda版本，升级到匹配版本，如果版本不匹配，即使编译成功会导致训练不收敛（亲测）。配置环境过程同https://blog.csdn.net/jdjjdjd/article/details/117731541
碰到的问题为：

编译opencv过程中报错，ippicv的包下载报错。由于翻墙的原因导致包下载出错，到opencv/3rdparty/ippicv/downloads/linux-808b791a6eac9ed78d32a7666804320e下载相应的tgz文件，目前环境下为ippicv_linux_20151201.tgz
编译caffe过程中报错，一个是c++11的错，需要在CMakeLists.txt第二行添加add_definitions(-std=c++11)即可。另一个错误是Makefile:588: recipe for target ‘.build_release/cuda/src/caffe/layers/softmax_loss_layer.o’ failed是需要更改Makefile中第409行。
NVCCFLAGS += -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)
改为：

NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)

Triton中文社区

欢迎来到由智源人工智能研究院发起的Triton中文社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。

更多推荐

torch.device(‘cuda:0‘ if cuda else ‘cpu‘)

# 这个地方的cuda:0 实际上并不是0号GPU，他取决于CUDA_VISIBLE_DEVICES# 然后逻辑GPU和物理GPU有一个对应关系# 如果CUDA_VISIBLE_DEVICES为2,1,3# 那么CUDA:0就是2号GPU， CUDA:1 就是1号GPU CUDA:3 就是3号GPUreturn torch.device('cuda:0' if cuda else 'cpu')..