Peixian02 commited on
Commit
972a3c6
·
verified ·
1 Parent(s): e1b0735

Match the environment line to the pinned cu118 stack

Browse files
Files changed (1) hide show
  1. README.md +5 -2
README.md CHANGED
@@ -97,8 +97,11 @@ python3 inference_bbx.py \
97
  default; add `--save_mask` for the full-resolution label mask and `--save_prob`
98
  for per-class probability maps.
99
 
100
- Environment: Python 3.12, torch 2.10.0+cu128, transformers 4.46.1,
101
- openslide-python with `openslide-bin`. See `environment.yml` in the GitHub repo.
 
 
 
102
 
103
  ## Citation
104
 
 
97
  default; add `--save_mask` for the full-resolution label mask and `--save_prob`
98
  for per-class probability maps.
99
 
100
+ Environment: Python 3.12, torch 2.4.0+cu118 with torchvision 0.19.0+cu118,
101
+ transformers 4.46.1, openslide-python with `openslide-bin`. The cu118 build covers
102
+ `sm_50` through `sm_90` (V100 up to H200); Blackwell cards (RTX 50 series, B200)
103
+ need a cu128 build instead - torch 2.10.0+cu128 with torchvision 0.25.0+cu128.
104
+ See `environment.yml` and the installation section of the README in the GitHub repo.
105
 
106
  ## Citation
107