pytorch1.7.1 cu110怎么安装
时间: 2025-02-23 19:29:47 浏览: 218
### 安装 PyTorch 1.7.1 和 CUDA 11.0
为了确保安装的是带有 CUDA 11.0 支持的 PyTorch 1.7.1 版本,可以按照如下方式进行操作:
对于使用 `pip` 的情况,推荐使用的命令为:
```bash
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple/
```
此命令指定了特定版本的 PyTorch 及其依赖项,并通过指定 URL 来获取支持 CUDA 11.0 的构建[^1]。
如果偏好使用 Conda 进行包管理,则应采用以下指令来创建新的 conda 环境并安装所需软件包:
```bash
conda create -n pytorch python=3.6
conda activate pytorch
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch
```
上述命令首先建立了一个名为 "pytorch" 的新环境,并设置了 Python 解释器版本;接着激活该环境后执行了针对 PyTorch 及关联库的具体版本与 CUDA 工具链匹配度良好的安装过程[^2]。
另外,在 Ubuntu 18.04 上准备相应的开发环境时,建议先完成 CUDA 11.0 和 cuDNN 的前置配置工作再继续以上步骤。这通常涉及下载 NVIDIA 提供的相关驱动程序以及设置必要的环境变量等准备工作[^3]。
阅读全文
相关推荐



















