vscode终端安装pytorch环境全流程小白版(linux+windows通用版)
在Windows和Linux系统中安装PyTorch环境的步骤大体相同,以下是一个简化的流程:
- 安装Python(如果尚未安装)。
- 使用pip安装PyTorch。
Windows系统
- 打开命令提示符(cmd)或PowerShell。
安装Python(如果尚未安装):
choco install python
或者去Python官网下载安装程序:https://www.python.org/downloads/
使用pip安装PyTorch:
pip install torch torchvision torchaudio
如果你使用的是特定版本的Python(如Python 3.x),请确保使用对应版本的pip。
Linux系统
- 打开终端。
安装Python(如果尚未安装):
sudo apt-get update sudo apt-get install python3
或者使用其他Linux发行版的包管理器安装Python。
使用pip安装PyTorch:
pip3 install torch torchvision torchaudio
或者如果你的系统中有多个Python版本,使用对应版本的pip命令,如
pip3.6
。
注意事项
- 在安装PyTorch之前,请确保你的网络连接正常,因为安装过程中会从网络下载文件。
- 如果你使用的是虚拟环境,请先激活你的虚拟环境再进行安装。
- 安装命令可能会根据你的具体需求(如CUDA版本)而有所不同,请参考PyTorch官方网站的安装指南:https://pytorch.org/get-started/locally/
如何检查PyTorch是否安装成功?
打开你的终端或命令提示符,输入以下Python代码检查PyTorch是否正确安装:
import torch
print(torch.__version__)
如果没有错误,并且打印出了PyTorch的版本号,则表示安装成功。
评论已关闭