如何在PyCharm远程开发中全局配置CUDA环境变量？

以下是三种可行的实现方式，覆盖不同场景需求：

方案1：修改远程服务器的用户shell配置（全局生效） #

如果团队成员各自拥有独立的服务器用户账号，这种方式最直接，所有该用户的会话（包括PyCharm远程连接）都会继承这个环境变量。

• 登录远程服务器，编辑对应shell的配置文件：
  • bash用户编辑~/.bashrc或~/.bash_profile
  • zsh用户编辑~/.zshrc
• 在文件末尾添加一行：

  export CUDA_VISIBLE_DEVICES=0
  

• 执行source ~/.bashrc（对应你的配置文件）让设置立即生效，后续重新连接PyCharm时会自动加载这个变量。

方案2：在PyCharm远程解释器中配置环境变量（项目专属） #

这种方式仅对当前PyCharm项目的远程调用生效，团队成员可各自在本地PyCharm中设置不同GPU设备号，互不干扰，最适合团队协作场景。

• 打开PyCharm，依次进入File > Settings > Project: [你的项目名] > Python Interpreter
• 点击远程解释器右侧的齿轮图标，选择Show All
• 在弹出列表中选中你的远程SSH解释器，点击Edit按钮
• 在编辑窗口中点击Environment Variables按钮
• 点击+号，添加环境变量：
  • 名称：CUDA_VISIBLE_DEVICES
  • 值：0（替换为你需要的设备号）
• 依次保存所有设置，之后该项目中所有远程运行的Python代码都会自动带上这个环境变量。

方案3：给Python解释器添加包装脚本（解释器级生效） #

如果需要固定某个Python解释器的所有调用都使用指定GPU，可以用包装脚本实现：

• 登录远程服务器，先找到Python解释器的路径：

  which python3
  

• 创建一个包装脚本，比如python3-gpu，内容如下（替换路径为你实际的Python路径）：

  #!/bin/bash
  export CUDA_VISIBLE_DEVICES=0
  exec /usr/bin/python3 "$@"
  

• 给脚本添加执行权限：

  chmod +x python3-gpu
  

• 在PyCharm的远程解释器配置中，将解释器路径替换为这个包装脚本的路径，之后所有通过该解释器运行的代码都会自动加载指定的环境变量。

内容的提问来源于stack exchange，提问作者Bram Vanroy