安装gpu版本的pytorch1.7.1
时间: 2025-02-01 18:08:17 浏览: 83
### 安装 PyTorch 1.7.1 GPU 版本
为了成功安装支持 CUDA 的 PyTorch 1.7.1,在 Windows 10 上的操作如下:
#### 准备工作
确保已确认设备所支持的 CUDA 版本,这可以通过查阅 NVIDIA 显卡官方网站获取显卡对应的驱动程序和支持的 CUDA 版本信息[^1]。
#### 创建并激活 Conda 虚拟环境
通过 Anaconda Prompt 来管理 Python 环境。首先删除旧有的可能冲突的虚拟环境(如果存在),接着创建一个新的名为 `pytorch_gpu` 的虚拟环境,并指定 Python 版本为 3.8:
```bash
conda remove -n pytorch_gpu --all
conda create -n pytorch_gpu python=3.8
```
激活新创建的虚拟环境:
```bash
conda activate pytorch_gpu
```
设置远程读取超时时间以防止网络不稳定引起的问题:
```bash
conda config --set remote_read_timeout_secs 2000.0
```
#### 安装 PyTorch 及其依赖项
根据目标硬件配置选择合适的 CUDA 工具包版本。对于 PyTorch 1.7.1 和 CUDA 11.0 组合而言,执行以下命令完成安装过程:
```bash
conda install pytorch cudatoolkit=11.0 -c pytorch
```
此操作会自动下载并安装与之匹配的 torchvision 和其他必要的组件[^4]。
#### 验证安装
最后验证安装是否成功以及确认安装的具体版本号:
```python
import torch
print(torch.__version__)
```
上述代码应该返回 "1.7.1" 并且能够检测到可用的 GPU 设备[^5]。
阅读全文
相关推荐


















