首页命令行下载pytorch

命令行下载pytorch

时间: 2024-12-28 21:20:29 浏览: 61

### 安装 PyTorch 的方法对于希望通过命令行安装 PyTorch 的用户，可以采用 `pip` 或者 Anaconda 来完成这一过程。 #### 使用 pip 安装 PyTorch CPU 版本可以直接在命令行工具中执行如下指令来安装适用于 CPU 的 PyTorch[^1]: ```bash pip install torch ``` #### 使用 pip 安装 PyTorch GPU 版本如果目标是安装支持 NVIDIA CUDA 技术的 GPU 加速版 PyTorch，则应依据具体的 CUDA 和 cuDNN 版本来选择合适的安装包。例如，为了安装特定版本（如 1.7.1），并兼容 CUDA 10.1 的 PyTorch 及其相关库，可按照下面的方式操作[^3]: ```bash pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html ``` 需要注意的是，在上述 URL 中包含了 `-f` 参数指定的一个链接地址，该链接指向了一个包含预编译二进制文件的位置列表页面，这有助于获取到与硬件环境相匹配的最佳配置组合。 #### 利用 Conda 创建新环境并安装带 GPU 支持的 PyTorch 另一种常见的做法是借助于 Anaconda 发行版所提供的虚拟环境管理功能创建一个新的 Python 环境，并在此环境中激活 GPU 加速特性：先建立名为 `pytorch_GPU` 的全新环境，并设定 Python 解释器版本为 3.9[^2]： ```bash conda create -n pytorch_GPU python=3.9 ``` 接着切换至刚创建好的环境内工作： ```bash conda activate pytorch_GPU ``` 之后便可以根据个人需求继续通过 `pip` 或其他方式进一步定制化安装所需的 PyTorch 组件及其依赖项。

阅读全文