Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

#

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Jump to bottom

Torch 2.0 fails with cuda=False #288

Closed
amontanez24 opened this issue May 5, 2023 · 0 comments · Fixed by #291
Closed

Torch 2.0 fails with cuda=False #288

amontanez24 opened this issue May 5, 2023 · 0 comments · Fixed by #291
Assignees
@amontanez24
Labels
bug Something isn't working
Milestone
0.7.3

Comments

@amontanez24
Copy link
Contributor

Environment Details

Please indicate the following details about the environment in which you found the bug:

  • CTGAN version: Master
  • Python version: Any
  • Operating System: Mac, Linux
  • Pytorch version: 2.0.0

Error Description

With Torch 2.0, the demo code fails whenever the cuda parameter is set to False. This is a problem because it is False by default in Linux.

Steps to reproduce

data = load_demo()
discrete_columns = [
    'workclass',
    'education',
    'marital-status',
    'occupation',
    'relationship',
    'race',
    'sex',
    'native-country',
    'income'
]

ctgan = CTGAN(epochs=10, cuda=False)
ctgan.fit(data, discrete_columns)

# Synthetic copy
samples = ctgan.sample(1000)
@amontanez24 amontanez24 added the bug Something isn't working label May 5, 2023
@amontanez24 amontanez24 mentioned this issue May 8, 2023
Merged
@amontanez24 amontanez24 added this to the 0.7.3 milestone May 25, 2023
@amontanez24 amontanez24 self-assigned this May 25, 2023
# for free to join this conversation on GitHub. Already have an account? # to comment
Assignees

@amontanez24 amontanez24

Labels
bug Something isn't working
Projects
None yet
Milestone
0.7.3
Development

Successfully merging a pull request may close this issue.

Fixing: Torch 2.0 fails with cuda=False sdv-dev/CTGAN
1 participant
@amontanez24