mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-19 05:54:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

small fixes

Browse Source
This commit is contained in:
Fangjun Kuang 2023-07-28 13:02:40 +08:00
parent 3271115acc
commit d75ffb7056
4 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
.github/workflows/run-docker-image.yml vendored
View File

@ -35,6 +35,10 @@ jobs:
nvcc --version nvcc --version
# For torch1.9.0-cuda10.2
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/compat:$LD_LIBRARY_PATH
which nvcc which nvcc
cuda_dir=$(dirname $(which nvcc)) cuda_dir=$(dirname $(which nvcc))
echo "cuda_dir: $cuda_dir" echo "cuda_dir: $cuda_dir"

2
docker/torch1.12.1-cuda11.3.dockerfile
View File

@ -1,4 +1,4 @@
FROM pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime FROM pytorch/pytorch:1.12.1-cuda11.3-cudnn8-devel
ENV LC_ALL C.UTF-8 ENV LC_ALL C.UTF-8

2
docker/torch1.9.0-cuda10.2.dockerfile
View File

@ -1,4 +1,4 @@
FROM pytorch/pytorch:1.9.0-cuda10.2-cudnn7-runtime FROM pytorch/pytorch:1.9.0-cuda10.2-cudnn7-devel
ENV LC_ALL C.UTF-8 ENV LC_ALL C.UTF-8

2
docker/torch2.0.0-cuda11.7.dockerfile
View File

@ -1,4 +1,4 @@
FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-devel
ENV LC_ALL C.UTF-8 ENV LC_ALL C.UTF-8