diff --git a/README.md b/README.md index 2a0e28d..a2f001c 100644 --- a/README.md +++ b/README.md @@ -11,6 +11,8 @@ A Python Library provides APIs to move PyTorch Tensors between CPU and NVMe. This package is only supported on Linux. `liburing` and `libaio` can be automatically installed. `liburing` is supported on Linux >= `5.10`, and it won't be installed if the version of your Linux < `5.10`. +> You must install pytorch and cmake before installing tensornvme. Once you upgrade pytorch, remember to reinstall tensornvme. + First, install requirements: ```shell pip install -r requirements.txt diff --git a/setup.py b/setup.py index ff30c5b..9bfa963 100644 --- a/setup.py +++ b/setup.py @@ -1,6 +1,5 @@ import os import sys -import torch from setuptools import setup, find_packages from torch.utils.cpp_extension import CppExtension, BuildExtension from subprocess import call @@ -85,9 +84,6 @@ def setup_dependencies(): def get_version(): with open('version.txt') as f: version = f.read().strip() - if sys.argv[1] != 'sdist': - torch_version = '.'.join(torch.__version__.split('.')[:2]) - version += f'+torch{torch_version}' return version