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()

成功!

相关推荐
向左转, 向右走ˉ9 分钟前
PyTorch随机擦除:提升模型抗遮挡能力
人工智能·pytorch·python·深度学习
七月初七淮水竹亭~14 小时前
Pycharm 报错 Environment location directory is not empty 如何解决
ide·python·pycharm
HuashuiMu花水木16 小时前
PyTorch笔记3----------统计学相关函数
人工智能·pytorch·笔记
AndrewHZ18 小时前
【图像处理基石】如何检测到画面中的ppt并对其进行增强?
图像处理·人工智能·pytorch·opencv·目标检测·计算机视觉·图像增强
June bug19 小时前
【python基础】python和pycharm的下载与安装
开发语言·python·pycharm
green5+121 小时前
yolo11n环境配置+实例运行【wsl+pycharm+conda】
ide·pycharm·conda
DAWN_T171 天前
Python编译器(Pycharm Jupyter)
python·机器学习·jupyter·pycharm
Draina1 天前
在pycharm中运行sagemath脚本的配置过程
ide·python·安全·pycharm·密码学
九章云极AladdinEdu1 天前
冷冻电镜重构的GPU加速破局:从Relion到CryoSPARC的并行重构算法
人工智能·pytorch·深度学习·机器学习·自然语言处理·架构·gpu算力
爱吃土豆的马铃薯ㅤㅤㅤㅤㅤㅤㅤㅤㅤ1 天前
${project.basedir}延申出来的Maven内置的一些常用属性
java·pycharm·maven