fix: неверный базовый образ для ROCm
This commit is contained in:
@@ -1,5 +1,8 @@
|
|||||||
# Use official Python image as base
|
# NVIDIA
|
||||||
FROM python:3.10-slim
|
#FROM python:3.10-slim
|
||||||
|
|
||||||
|
# AMD
|
||||||
|
FROM rocm/pytorch:latest
|
||||||
|
|
||||||
# Set working directory
|
# Set working directory
|
||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
|
|||||||
Reference in New Issue
Block a user