pycharm添加虚拟环境以及虚拟环境安装pytorch

  1. file、settings、interpreter、add interpreter、add local interpreter

    记住不要勾选inherit,不然会把主环境的东西继承到虚拟环境。

    创建前可以先点existing看看有没有已经建好的虚拟环境

  2. 有的时候pycharm有问题,创建了虚拟环境没有显示。找一个.py文件,上方会弹出提醒,选择使用interpreter,选择刚刚建好的虚拟环境。

  3. 记得重新打开terminal,如果前面有(venv),说明是在虚拟环境中。

  4. pytorch官网,pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

  5. 运行一段代码,调用gpu看看行不行

    import torch

    if torch.cuda.is_available():
    print("检测到当前设备有可用GPU:")
    print("当前可用GPU数量:", torch.cuda.device_count())
    print("当前GPU索引:", torch.cuda.current_device())
    print("当前GPU名称:", torch.cuda.get_device_name(torch.cuda.current_device()))
    else:
    print("未检测到当前设备有可用GPU,不建议开始训练,如有需求请自行更改代码:")
    exit()

成功!

相关推荐
MYX_3091 小时前
第四章 神经网络的基本组件
pytorch·深度学习·神经网络·学习
新手村领路人2 小时前
Ubuntu18.04 libc.so.6: version `GLIBC_2.28‘ not found问题
linux·ubuntu·pycharm
西柚小萌新3 小时前
【深入浅出PyTorch】--7.2.PyTorch可视化2
人工智能·pytorch·python
Sylvan Ding4 小时前
VSCode插件推荐 2025 - 拥抱 Agentic Coding 时代:是时候从 PyCharm 切换到 VSCode 生态了!
ide·vscode·pycharm·extension·插件·agentic·氛围编程
七夜zippoe7 小时前
大显存 AI 训练实战:PyTorch/TensorFlow 参数调试与多场景落地指南
人工智能·pytorch·深度学习
LiJieNiub7 小时前
基于 PyTorch 的图像分类模型集成实践
人工智能·pytorch·分类
芥子沫7 小时前
PyTorch是什么?
人工智能·pytorch
叶落阁主10 小时前
IdeaVim 新插件推荐- Vim CmdFloat 优化搜索/命令执行框
pycharm·vim·intellij idea
ZHOU_WUYI12 小时前
LLMs-from-scratch :embeddings 与 linear-layers 的对比
pytorch·python·llm
雲_kumo12 小时前
从零开始读懂Transformer:架构解析与PyTorch实现
pytorch·架构·transformer