无需 anaconda,只使用 pip 也可以在 PyCharm 集成环境中配置深度学习 PyTorch。
本文全部信息及示范来自 PyTorch 官网。

以防你是super小白:
PyCharm 中的命令是在 Python Terminal 中运行,界面左下角竖排图标第四个。

1. 安装前置包 numpy

pip install numpy

2. 安装 PyTorch

安装命令根据官网指示获取:
https://pytorch.org/get-started/locally/

举例:我的电脑为 Windows 系统,我决定使用 Pip 安装,在 Python 中使用,电脑无独立显卡(仅CPU),那么我的选择如下图所示:
在这里插入图片描述
复制最下面一行的命令,即:

pip3 install torch torchvision torchaudio

到 PyCharm 中运行。等待下载与安装,即可完成。

验证示例:

新建 .py 文件,输入:

import torch
x = torch.rand(5, 3)
print(x)

输出例如:

D:\etude\cs\python\pythonProjects\Mar22\.venv\Scripts\python.exe D:\etude\cs\python\pythonProjects\Mar22\.venv\test.py 
tensor([[0.3016, 0.0369, 0.9889],
        [0.1310, 0.4814, 0.9994],
        [0.2568, 0.5121, 0.1174],
        [0.3562, 0.0751, 0.1503],
        [0.2874, 0.3376, 0.6854]])

Process finished with exit code 0

即证明 PyTorch 安装成功。

好消息

目前 PyTorch 最新版 2.6 已经可以支持 Python 最新版 3.13
无心插柳柳成荫,在自作主张升级了最新版 Python 并顺利安装 PyTorch 后,发现之前很多人控诉版本不适配的问题。
在这里插入图片描述

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐