MVPilgrim commited on
Commit
b1bc691
·
1 Parent(s): 03da657
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -3,8 +3,8 @@ FROM nvidia/cuda:12.2.0-base-ubuntu22.04 AS base
3
 
4
  ENV DEBIAN_FRONTEND=noninteractive
5
  ENV DEBCONF_NOWARNINGS="yes"
6
- ENV CUDA_VISIBLE_DEVICES=0
7
- ENV LLAMA_CUBLAS=1
8
 
9
  # Install necessary dependencies and musl
10
  RUN apt-get update && apt-get install -y \
 
3
 
4
  ENV DEBIAN_FRONTEND=noninteractive
5
  ENV DEBCONF_NOWARNINGS="yes"
6
+ #ENV CUDA_VISIBLE_DEVICES=0
7
+ #ENV LLAMA_CUBLAS=1
8
 
9
  # Install necessary dependencies and musl
10
  RUN apt-get update && apt-get install -y \