search

BiomedSciAI / fuse-med-ml

A python framework accelerating ML based discovery in the medical field by encouraging code reuse. Batteries included :)
Apache License 2.0
137 stars 34 forks source link

Fix unitests with latest pytorch #136

Closed mosheraboh closed 2 years ago

mosheraboh commented 2 years ago
  1. Fixing unitest issue: "Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you must use the 'spawn' start method""
  2. Upgrading pipeline to CUDA 11.6 to fit all GPUs in jenkins
  3. Fix black and mypy version to ensure stable checkers
  4. Move some dependencies to requiements_dev.txt