PyCharm中torch.cuda.is_available()返回True,终端中返回False
解决终端中PyTorch CUDA不可用但PyCharm中可用的问题
排查与解决步骤
确认终端激活的conda环境正确性
- 运行
conda info --envs查看当前激活环境(带*标记项),确保和PyCharm使用的环境一致。 - 执行
which python,将输出路径和PyCharm项目配置里的Python解释器路径对比。如果不一致,重新执行conda activate 你的环境名切换到目标环境。
- 运行
检查终端的CUDA环境变量
PyCharm会自动加载conda环境的完整变量,但终端可能缺失关键路径:- 激活环境后,运行
echo $PATH,确认包含conda环境的bin目录(如/path/to/conda/envs/your_env/bin)。 - 运行
echo $LD_LIBRARY_PATH,检查是否包含PyTorch依赖的CUDA库路径(通常是/path/to/conda/envs/your_env/lib/pythonX.X/site-packages/torch/lib)。 - 若路径缺失,手动添加:
可将这两行添加到shell配置文件(如export LD_LIBRARY_PATH=/path/to/conda/envs/your_env/lib:$LD_LIBRARY_PATH export PATH=/path/to/conda/envs/your_env/bin:$PATH~/.bashrc或~/.zshrc),避免每次打开终端重复设置。
- 激活环境后,运行
验证PyTorch的CUDA版本一致性
在终端和PyCharm中分别运行:import torch print(torch.version.cuda)如果版本不一致,说明终端未调用目标环境的PyTorch。重新激活环境后确认
which python路径正确,必要时重新执行安装命令:conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia检查GPU驱动的终端访问情况
运行nvidia-smi,若无法显示GPU信息,可能是终端未加载驱动环境变量。但如果PyCharm能正常访问,优先排查conda环境变量问题。
内容的提问来源于stack exchange,提问作者Khawar Islam
相关产品推荐
相关产品推荐

