adding example

pytorch-rust-docker/chatgpt-pytorch/Cargo.toml

@@ -0,0 +1,10 @@
+[package]
+name = "chatgpt-pytorch"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+tch = "0.11.0"
+

pytorch-rust-docker/chatgpt-pytorch/Dockerfile

@@ -0,0 +1,18 @@
+# Use the official Rust image as the base image
+FROM rust:1.67
+
+# Set the working directory
+WORKDIR /app
+
+# Install system dependencies required by PyTorch and tch-rs
+RUN apt-get update && \
+    apt-get install -y build-essential python3-dev python3-pip libopenblas-dev libopencv-dev
+
+# Copy the Rust project into the container
+COPY . .
+
+# Build the Rust project
+RUN cargo build --release
+
+# Set the command to run the binary
+CMD ["./target/release/chatgpt_pytorch"]

pytorch-rust-docker/chatgpt-pytorch/src/main.rs

@@ -0,0 +1,6 @@
+use tch::{Tensor, Kind};
+
+fn main() {
+    let tensor = Tensor::of_slice(&[1, 2, 3, 4]).reshape(&[2, 2]).to_kind(Kind::Float);
+    println!("Tensor: {:?}", tensor);
+}