diff --git a/.gitattributes b/.gitattributes
index 26e3729..fffc3c4 100644
--- a/.gitattributes
+++ b/.gitattributes
@@ -1 +1,2 @@
 src/_version.py export-subst
+src/rembg/_version.py export-subst
diff --git a/Dockerfile b/Dockerfile
index 2e023f4..0829843 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.4.1-devel-ubi8
+FROM nvidia/cuda:11.4.2-cudnn8-runtime-ubuntu20.04
 
 RUN apt-get update && apt-get install -y --no-install-recommends python3 python3-pip python3-dev llvm llvm-dev
 RUN pip3 install rembg
diff --git a/MANIFEST.in b/MANIFEST.in
index d0d0961..99fe2d3 100644
--- a/MANIFEST.in
+++ b/MANIFEST.in
@@ -12,3 +12,4 @@ recursive-include data *
 include requirements.txt
 include versioneer.py
 include src/_version.py
+include src/rembg/_version.py
diff --git a/pyproject.toml b/pyproject.toml
index c0553ca..b90ff7c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -7,6 +7,6 @@ build-backend = "setuptools.build_meta"
 [versioneer]
 VCS = "git"
 style = "default"
-versionfile_source = "src/_version.py"
+versionfile_source = "src/rembg/_version.py"
 versionfile_build = "_version.py"
 tag_prefix = ""
diff --git a/setup.cfg b/setup.cfg
index bc98b51..63211d6 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -10,6 +10,6 @@ license_files = LICENSE.txt
 [versioneer]
 VCS = git
 style = default
-versionfile_source = src/_version.py
+versionfile_source = src/rembg/_version.py
 versionfile_build = _version.py
 tag_prefix =
diff --git a/src/rembg/__init__.py b/src/rembg/__init__.py
index e69de29..ecd3379 100644
--- a/src/rembg/__init__.py
+++ b/src/rembg/__init__.py
@@ -0,0 +1,3 @@
+
+from . import _version
+__version__ = _version.get_versions()['version']
diff --git a/src/_version.py b/src/rembg/_version.py
similarity index 99%
rename from src/_version.py
rename to src/rembg/_version.py
index 2ab7833..e886948 100644
--- a/src/_version.py
+++ b/src/rembg/_version.py
@@ -44,7 +44,7 @@ def get_config():
     cfg.style = "default"
     cfg.tag_prefix = ""
     cfg.parentdir_prefix = "None"
-    cfg.versionfile_source = "src/_version.py"
+    cfg.versionfile_source = "src/rembg/_version.py"
     cfg.verbose = False
     return cfg