Skip to content

Commit

Permalink
fix version
Browse files Browse the repository at this point in the history
  • Loading branch information
feifeibear committed Jul 19, 2022
2 parents f6e6576 + eaa20fa commit b754a66
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 4 deletions.
2 changes: 2 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
4 changes: 0 additions & 4 deletions setup.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down Expand Up @@ -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


Expand Down

0 comments on commit b754a66

Please sign in to comment.