liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-17 09:24:34 +08:00
Code Activity

Fix GPU

Browse Source
This commit is contained in:
Jactus
2020-12-02 12:23:42 +08:00
parent a2603fe27a
commit d109d3d44e
6 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
qlib/contrib/model/pytorch_nn.py
View File

@@ -151,7 +151,7 @@ class DNNModelPytorch(Model):
self.dnn_model.cuda()
# set the visible GPU
if self.visible_GPU:
os.environ["CUDA_VISIBLE_DEVICES"] = self.visible_GPU
os.environ["CUDA_VISIBLE_DEVICES"] = str(self.visible_GPU)
def fit(
self,