update

Dockerfile.bgpt

@@ -21,9 +21,23 @@ echo "PubkeyAuthentication yes" >> /etc/ssh/sshd_config
 echo "Port 22" >> /etc/ssh/sshd_config
 mkdir /var/run/sshd
 echo 'root:cdcdocker' | chpasswd
-# Install Micromamba
+# 安装 micromamba 并配置 mambarc
 echo 1 | bash <(curl -s https://cdn.jsdelivr.net/gh/hotwa/MicroMamba_Installer@main/install.sh)
 micromamba shell init -s bash -p ~/micromamba
+cat <<'EOF' >> ~/.bashrc
+source ~/micromamba/etc/profile.d/micromamba.sh
+alias mamba=micromamba
+alias mba=mamba
+EOF
+# 配置 .mambarc 文件
+cat <<EOF > ~/.mambarc
+channels:
+  - conda-forge
+  - bioconda
+  - pytorch
+  - pytorch-nightly
+  - nvidia
+EOF
 mkdir -p ~/.pip
 echo "
 [global]