win10+conda+pytorch-gpu安装

本文档介绍了在Windows 10上通过Anaconda安装PyTorch-GPU的详细步骤,包括下载Anaconda、设置conda环境、安装CUDA和cuDNN,以及验证安装成功的测试代码。提供了适用于不同GPU环境的安装指南和相关资源链接。

pytorch 官网上只有linux和Mac的程序包,没有windows系统的,但是windows系统还是可以用pytorch的。 
github: https://github.com/peterjc123/pytorch-scripts

首先要安装anaconda:

推荐清华镜像 
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ 
拖到最后,下载最新版即可。

这里写图片描述

安装完毕要安装pytorch运行环境:

# If your main Python version is not 3.5 or 3.6
conda create -n pytorch python=3.6 numpy pyyaml mkl
  • 3. activate pytorch36 激活这个虚拟环境(取消激活用deactivate)

    4. 在pytorch 官网上寻找自己的环境,注意自己电脑是不是有显卡,查看自己是否是独显还是集成显卡(集成显卡就是核显)在这里不能使用cuda 安装,按win+R 输入dxdiag 找到显示,查看显卡属性,然后百度搜是否是集成显卡还是独显

### 安装准备 为了在 Windows 10 上成功离线安装 PyTorch GPU 版本,需先准备好必要的文件并设置好环境。确保已下载适用于目标系统的 CUDA 工具包以及 cuDNN 库,并确认其版本兼容所要安装PyTorch 版本。 #### 下载所需组件 - **CUDA Toolkit**: 根据所需的 PyTorch 和 CUDA 的匹配关系选择合适的 CUDA 版本。对于 PyTorch 2.1.0 推荐使用 CUDA 11.8[^2]。 - **cuDNN Library**: 需与选定的 CUDA 版本相适配。可以从 NVIDIA 官方网站获取相应版本的 cuDNN 文件。 - **Miniconda 或 Anaconda**: 使用 Miniconda 创建 Python 虚拟环境来管理依赖项更为简便。推荐选用带有 Python 3.8 及以上版本的 Miniconda 发行版[^3]。 - **PyTorch 相关 whl 文件**: 访问官方提供的离线资源页面或通过其他可靠渠道提前下载适合本地硬件条件(如特定 CUDA 版本)的 `torch`, `torchvision` 和 `torchaudio` 的 `.whl` 文件[^4]。 ### 设置开发环境 完成上述软件包收集之后,在未联网的情况下按照如下步骤操作: #### 初始化 Conda 环境 启动命令提示符窗口,执行以下指令初始化一个新的 conda 环境用于承载即将部署的应用程序及其依赖库: ```bash conda create -n pytorch_env python=3.8 conda activate pytorch_env ``` #### 手动添加路径至系统环境变量 将之前提到过的 CUDA 组件所在目录加入到系统的 PATH 中以便于后续调用编译器和其他工具链功能。具体来说就是把下面这些位置追加进去(假设默认安装到了标准路径下): - `C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin` - `C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\include` - `C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\libnvvp` 这一步骤可以参照常规方法修改注册表或者利用图形界面来进行调整[^1]。 #### 安装预下载好的 Whl 包 切换回激活状态下的 conda 环境,依次运行下列 pip 命令以加载先前保存下来的 wheel 文件: ```bash pip install torch-<version>-win_amd64.whl pip install torchvision-<version>-win_amd64.whl pip install torchaudio-<version>-win_amd64.whl ``` 请注意替换 `<version>` 占位符为实际文件名中的版本号部分。 最后验证安装成果可通过打开 Python 解释器输入 `import torch;print(torch.cuda.is_available())` 来查看返回值是否为 True 表明一切正常工作。
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

一摩尔自由

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值