PyTorch环境变量配置

PyTorch环境中有一些关键的变量用于配置其运行环境，这些变量通常在`torch`模块下或者通过命令行设置。以下是几个重要的环境变量：

1. **CUDA_VISIBLE_DEVICES**: 如果你有多个GPU，这个变量可以帮助指定哪些设备应该被PyTorch看到。例如，`export CUDA_VISIBLE_DEVICES=0,1`会将第0和1号GPU设置为默认。

2. **TORCH_HOME** 或 **PYTORCH_HOME**: 这些变量指定了PyTorch库的安装位置，特别是对于下载的数据和模型文件。你可以自定义它来管理你的存储路径。

3. **TORCH_DEVICE**: 这个变量可以手动设置默认的计算设备，如CPU (`cpu`) 或者GPU (`cuda`). 默认情况下，如果检测到可用的GPU，它会被自动设置为GPU。

4. **PYTHONPATH**: 包含额外的Python模块搜索路径，有助于找到PyTorch及其相关的库。

5. **CUDA_PATH**: 对于CUDA版本的管理，如果你不是默认安装CUDA，可以设置这个环境变量指向CUDA的安装目录。

6. **LD_LIBRARY_PATH**: 用于动态链接库查找路径，确保PyTorch能够找到依赖的库文件。

为了设置这些变量，通常在终端中使用`export`命令，或者在bash、zsh等 shell脚本中添加它们。如果你想让这些设置永久生效，可以在系统的`~/.bashrc` 或 `~/.bash_profile` 文件中添加相应的配置。