在官方文档(https://pytorch.org/get-started/locally/)中显示:目前,Windows 上的 PyTorch 仅支持 Python 3.7-3.9;不支持 Python 2.x。
因此可根据个人需求选择 3.7-3.9 的版本
若电脑为集显,或显卡为AMD,或NVIDIA显卡版本较低直接选择CPU即可,因此我们的环境为:Stable(1.10.2),Canda,Python,CPU
对应的命令为:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
其他环境需求可看:https://pytorch.org/get-started/locally/
为了方便管理我们的项目,我们首先在Anaconda中创建一个虚环境
创建python版本为3.8,名为“pytorch”的虚环境
conda create --name pytorch python=3.8
activate pytorch
以下为CPU版的命令,其他版本可参考(https://pytorch.org/get-started/locally/)
conda install pytorch torchvision torchaudio cpuonly -c pytorch
在命令行中输入:
python
输入:
import torch
x = torch.rand(5, 3)
print(x)