Windows下配置PyTorch环境的简单方法和步骤如下:
1. 安装Python
需要在Windows系统上安装Python环境,推荐使用Anaconda,可以从官网下载最新版本的Python,并安装,安装过程中会提示添加环境变量,一定要勾选,以便后续使用Python命令。
2. 安装PyTorch
安装PyTorch之前,需要检查当前系统的Python版本,以及是否支持CUDA(如果支持,需要安装对应版本的CUDA),打开Anaconda Prompt,输入以下命令:
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
安装完成后,可以输入以下命令,查看是否安装成功:
python -c "import torch;print(torch.__version__)"
3. 安装PyTorch依赖库
安装完PyTorch之后,还需要安装一些PyTorch依赖库,例如NumPy、Matplotlib、Pandas等,可以使用pip命令或者conda命令进行安装,例如:
pip install numpy conda install matplotlib pip install pandas
4. 测试PyTorch环境
可以使用以下代码进行简单的测试,以确认PyTorch环境是否配置成功:
import torch x = torch.rand(5, 3) print(x)
如果能正常输出结果,则表示PyTorch环境配置成功!