!curl -O https://raw.githubusercontent.com/deepjavalibrary/d2l-java/master/tools/fix-colab-gpu.sh && bash fix-colab-gpu.sh
Since Java is not natively supported by Colab, we need to run the following code to enable Java kernel on Colab.
Since Colab supports CUDA 10.1, we will have to follow some steps to setup the environment. Refresh the page (press F5) and stay at Python runtime on GPU. Run the file fix-colab-gpu script.
And then ensure that you have switched to CUDA 10.1. 3. After that, switch runtime to Java and hardware to GPU.(Might require refreshing the page and switching runtime)
Now you can write Java code.
!curl -O https://raw.githubusercontent.com/deepjavalibrary/d2l-java/master/tools/colab_build.sh && bash colab_build.sh
%maven ai.djl:basicdataset:0.23.0
%maven ai.djl.mxnet:mxnet-engine:0.23.0
%maven org.slf4j:slf4j-simple:1.7.36
import ai.djl.*;
import ai.djl.engine.*;
import ai.djl.inference.*;
import ai.djl.modality.*;
import ai.djl.modality.cv.*;
import ai.djl.modality.cv.Image;
import ai.djl.modality.cv.transform.*;
import ai.djl.modality.cv.translator.*;
import ai.djl.modality.cv.output.*;
import ai.djl.modality.cv.util.*;
import ai.djl.ndarray.*;
import ai.djl.ndarray.index.*;
import ai.djl.ndarray.types.*;
import ai.djl.ndarray.types.Shape;
import ai.djl.nn.*;
import ai.djl.nn.convolutional.*;
import ai.djl.nn.core.*;
import ai.djl.nn.norm.*;
import ai.djl.nn.pooling.*;
import ai.djl.nn.recurrent.*;
import ai.djl.repository.zoo.*;
import ai.djl.training.*;
import ai.djl.training.evaluator.*;
import ai.djl.training.initializer.*;
import ai.djl.training.listener.*;
import ai.djl.training.loss.*;
import ai.djl.training.optimizer.*;
import ai.djl.training.tracker.*;
import ai.djl.training.util.*;
import ai.djl.training.dataset.*;
import ai.djl.translate.*;
import ai.djl.util.*;
import java.io.*;
import java.net.*;
import java.nio.*;
import java.nio.file.*;
import java.util.*;
import java.util.function.*;
String datasetLimit = System.getenv("DATASET_LIMIT");
if (datasetLimit != null) {
System.setProperty("DATASET_LIMIT", datasetLimit);
}
String maxEpoch = System.getenv("MAX_EPOCH");
if (maxEpoch != null) {
System.setProperty("MAX_EPOCH", maxEpoch);
}
System.setProperty("org.slf4j.simpleLogger.showThreadName", "false");
System.setProperty("org.slf4j.simpleLogger.showLogName", "false");
System.setProperty("org.slf4j.simpleLogger.log.ai.djl.pytorch", "WARN");
System.setProperty("org.slf4j.simpleLogger.log.ai.djl.mxnet", "WARN");
System.setProperty("org.slf4j.simpleLogger.log.ai.djl.tensorflow", "WARN");