From bae8df8b0f276dcdd283d7be40f28862ce061ac3 Mon Sep 17 00:00:00 2001
From: Fangjun Kuang <csukuangfj@gmail.com>
Date: Thu, 27 Jul 2023 19:07:53 +0800
Subject: [PATCH] small fixes

---
 .github/workflows/build-docker-image.yml      | 3 +++
 docker/torch1.13.0-cuda11.6-cudnn8.dockerfile | 4 ++++
 2 files changed, 7 insertions(+)

diff --git a/.github/workflows/build-docker-image.yml b/.github/workflows/build-docker-image.yml
index 1a905c4de..2e07a3e4d 100644
--- a/.github/workflows/build-docker-image.yml
+++ b/.github/workflows/build-docker-image.yml
@@ -2,6 +2,9 @@
 # https://docs.github.com/en/actions/publishing-packages/publishing-docker-images#publishing-images-to-github-packages
 name: Build docker image
 on:
+  push:
+    branches:
+      - docker2
   workflow_dispatch:
 
 concurrency:
diff --git a/docker/torch1.13.0-cuda11.6-cudnn8.dockerfile b/docker/torch1.13.0-cuda11.6-cudnn8.dockerfile
index e6afea18b..dd95ce441 100644
--- a/docker/torch1.13.0-cuda11.6-cudnn8.dockerfile
+++ b/docker/torch1.13.0-cuda11.6-cudnn8.dockerfile
@@ -44,11 +44,15 @@ RUN pip install --no-cache-dir \
       typeguard \
       dill
 
+RUN cd /opt/conda/lib/stubs && ln -s libcuda.so libcuda.so.1
+
 RUN git clone https://github.com/k2-fsa/icefall /workspace/icefall && \
 	cd /workspace/icefall && \
 	pip install --no-cache-dir -r requirements.txt
 
 ENV PYTHONPATH /workspace/icefall:$PYTHONPATH
 
+ENV LD_LIBRARY_PATH /opt/conda/lib/stubs:$LD_LIBRARY_PATH
+
 WORKDIR /workspace/icefall