Abhishek Thakur
		
	commited on
		
		
					Commit 
							
							·
						
						e2ff59b
	
1
								Parent(s):
							
							e65f8ef
								
remove cuda toolkit
Browse files- Dockerfile +2 -2
    	
        Dockerfile
    CHANGED
    
    | @@ -64,8 +64,8 @@ SHELL ["conda", "run","--no-capture-output", "-p","/app/env", "/bin/bash", "-c"] | |
| 64 |  | 
| 65 | 
             
            RUN conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia && \
         | 
| 66 | 
             
                conda clean -ya && \
         | 
| 67 | 
            -
                conda install -c "nvidia/label/cuda-12.1.0" cuda-nvcc && conda clean -ya | 
| 68 | 
            -
             | 
| 69 |  | 
| 70 | 
             
            COPY --chown=1000:1000 . /app/
         | 
| 71 | 
             
            RUN make socket-kit.so
         | 
|  | |
| 64 |  | 
| 65 | 
             
            RUN conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia && \
         | 
| 66 | 
             
                conda clean -ya && \
         | 
| 67 | 
            +
                conda install -c "nvidia/label/cuda-12.1.0" cuda-nvcc && conda clean -ya
         | 
| 68 | 
            +
            # conda install -c "nvidia/label/cuda-12.1.0" cuda-toolkit && conda clean -ya
         | 
| 69 |  | 
| 70 | 
             
            COPY --chown=1000:1000 . /app/
         | 
| 71 | 
             
            RUN make socket-kit.so
         | 
