Closed icsl-Jeon closed 1 year ago
Describe the bug
Thank you for wonderful OpenSource project!
The installation guide here, which uses
pip install .
, can install pytorch cpu version. If this is the case, users cannot use gpu when training.I think it would be better to instruct users that the below pytorch install might be required
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
To Reproduce We can simply check the below after installing FinRL on a clean conda env.
import torch torch.cuda.is_available()
Thanks very much for this feedback!
@icsl-Jeon thanks for your suggestions w.r.t torch, it depends on users' demand, e.g., OS, gpu/cpu the install docs focus on installing finrl
Describe the bug
Thank you for wonderful OpenSource project!
The installation guide here, which uses
pip install .
, can install pytorch cpu version. If this is the case, users cannot use gpu when training.I think it would be better to instruct users that the below pytorch install might be required
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
To Reproduce We can simply check the below after installing FinRL on a clean conda env.