把torchGPU加到pycharm上
时间: 2023-09-21 09:06:34 浏览: 71
Anaconda+spyder+pycharm的pytorch配置详解(GPU)
首先,确保你已经安装了PyTorch和CUDA,而且你的PyCharm也已经正确安装。
接下来,你需要按照以下步骤将torchGPU添加到PyCharm中:
1. 打开PyCharm,点击File -> Settings -> Project -> Project Interpreter
2. 点击右上角的“+”按钮,搜索并选择torch和torchvision的GPU版本(例如:torch==1.8.0+cu111)
3. 安装选中的库
4. 在PyCharm的Run configurations中设置CUDA_VISIBLE_DEVICES环境变量,该变量的值应该是你要使用的GPU的编号。例如,如果你要使用第二个GPU,则设置为“1”。
现在,你就可以在PyCharm中使用torchGPU了。
阅读全文