获取 PyTorch
首先,需要设置 Python 环境。
建议使用 Anaconda 以包管理员身份在 Windows 中设置虚拟 Python 环境。 此设置的其余部分假定你使用 Anaconda 环境。
1.在此处下载并安装 Anaconda。 选择 Anaconda 64-bit installer for Windows Python 3.9。安装最新版本即可
注意:一定要安装Python 3.x,因为PyTorch不支持Python2.x
2.安装完成后,请验证 Anaconda 和 Python 版本。
可通过运行以下命令来检查你的 Python 版本:python –-version
可通过运行以下命令来检查你的 Anaconda 版本:conda –-version
接下来就可以正常安装PyTorch了。
3.导航到 https://pytorch.org/
选择相关的 PyTorch 安装详细信息:
PyTorch 版本 – 稳定版。
你的 OS – Windows 包 –
Conda 语言 – Python
计算平台 – CPU,或者选择 Cuda 的版本。
4.打开 Anaconda 管理器,并运行安装说明中指定的命令。
conda install pytorch torchvision torchaudio cpuonly -c pytorch
输入该指令后anaconda就会自动安装PyTorch。
5.稍等片刻即可安装成功,输入以下指令测试是否成功配置
>>python
>>import torch
>>x=torch.rand(2, 3)
>>print(x)
查看输出是否是2*3的二维表且数字随机。
如果成功显示那么重启vscode即可完成PyTorch的配置。
文章出处登录后可见!
已经登录?立即刷新