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

成功!

相关推荐
科大饭桶10 小时前
昇腾AI自学Day2-- 深度学习基础工具与数学
人工智能·pytorch·python·深度学习·numpy
weixin_5079299112 小时前
第G7周:Semi-Supervised GAN 理论与实战
人工智能·pytorch·深度学习
weixin_4569042714 小时前
一文讲清楚Pytorch 张量、链式求导、正向传播、反向求导、计算图等基础知识
人工智能·pytorch·学习
充气大锤16 小时前
从0开始配置conda环境并在PyCharm中使用
ide·pycharm·conda
盼小辉丶21 小时前
Transformer实战(4)——从零开始构建Transformer
pytorch·深度学习·transformer
之歆1 天前
Al大模型-本地私有化部署大模型-大模型微调
人工智能·pytorch·ai作画
失散131 天前
深度学习——02 PyTorch
人工智能·pytorch·深度学习
Re_draw_debubu1 天前
神经网络 小土堆pytorch记录
pytorch·神经网络·小土堆
小徐敲java2 天前
python-pycharm切换python各种版本的环境与安装python各种版本的环境(pypi轮子下载)
开发语言·python·pycharm
coding者在努力2 天前
从零开始:用PyTorch实现线性回归模型
人工智能·pytorch·线性回归