docker for inference, works on linux and windows

This commit is contained in:
jason-on-salt-a40
2024-03-30 11:32:09 -07:00
parent c54b343bfb
commit 741a6559e9
4 changed files with 80 additions and 62 deletions

View File

@@ -8,7 +8,8 @@
"source": [
"import os\n",
"os.environ[\"CUDA_DEVICE_ORDER\"]=\"PCI_BUS_ID\" \n",
"os.environ[\"CUDA_VISIBLE_DEVICES\"]=\"7\""
"os.environ[\"CUDA_VISIBLE_DEVICES\"]=\"7\"\n",
"os.environ[\"USER\"] = \"YOUR_USERNAME\" # TODO change this to your username"
]
},
{