update docker

pull/4072/head
zxcd 4 months ago
parent 0394ac59b1
commit 463ec04ad5

@ -174,7 +174,7 @@ class AttLoc(nn.Layer):
att_prev = att_prev / enc_hs_len.unsqueeze(-1).astype(
att_prev.dtype)
else:
att_prev.unsqueeze(-1) # only for static graph
att_prev.unsqueeze(-1) # only for static graph
# att_prev: (utt, frame) -> (utt, 1, 1, frame)
# -> (utt, att_conv_chans, 1, frame)

@ -1,4 +1,4 @@
FROM registry.baidubce.com/paddlepaddle/paddle:latest-dev-cuda11.2-cudnn8.2-trt8.0-gcc82
FROM registry.baidubce.com/paddlepaddle/paddle:latest-dev-cuda12.3-cudnn9.0-trt8.6-gcc12.2
RUN apt-get update -y
RUN apt-get -y install libsndfile1 lsof
RUN pip3.8 install pytest-runner
RUN pip3.10 install pytest-runner

Loading…
Cancel
Save