skip to Main Content

Using GPU in VS code container – Docker

I want to use a GPU inside a Visual Studio Code docker container to train model with TensorFlow. To build an image for my container I use next Dockerfile: FROM mcr.microsoft.com/vscode/devcontainers/anaconda:0-3 ARG PROJECT_NAME=fire_rec ARG NODE_VERSION="none" RUN if [ "${NODE_VERSION}" !=…

VIEW QUESTION
Back To Top
Search