We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2c6b561 commit a5b777cCopy full SHA for a5b777c
docs/installation.md
@@ -56,8 +56,7 @@ the top of your general simulation scripts as follows:
56
mid = -1 # the gpu_id (run nivida-smi to find your system's GPU identifiers)
57
if mid >= 0:
58
print(" > Using GPU ID {0}".format(mid))
59
- os.environ["CUDA_VISIBLE_DEVICES"]="{0}".format(mid)
60
- #gpu_tag = '/GPU:0'
+ os.environ["CUDA_VISIBLE_DEVICES"]="{0}".format(mid)'
61
gpu_tag = '/GPU:0'
62
else:
63
os.environ["CUDA_VISIBLE_DEVICES"]="-1"
0 commit comments