win10安装pytorch

本文详细介绍了如何在Anaconda环境下安装PyTorch的过程,包括配置镜像源、创建虚拟环境、安装对应CUDA版本的PyTorch及验证安装是否成功。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1、安装pytorch需要提前安装好anaconda

配置anaconda镜像源:

      conda config --add channel http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/

       conda config --add channel  http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/

       conda config --add cannel http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64

       conda config --set show_channel_urls yes

查看当前使用源:

conda config --show-sources

运行 conda clean -i 清除索引缓存,保证用的是镜像站提供的索引。

还原原始源:conda config --remove-key channels

2、打开 Anaconda 自带的 Anaconda Prompt:创建虚拟环境

先查看安装的python版本:

conda create --name pytorch python=3.9

注意,这里的 pytorch 是虚拟环境的名称,可随意取。

创建好后会提示:

3、进入pytorch虚拟环境:

activate pytorch

退出虚拟环境:

conda deactivate

4、安装pytorch

4.1 先查看当前计算机的cuda版本(英伟达版本):nvidia-smi

4.2 选择对应的pytorch版本安装

Start Locally | PyTorch

注意:由于cuda版本是11.4在这里就只能选cuda 11.3的版本进行安装

安装命令:

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

注意:如果安装过程中报HTTP 000 之类的错误,可以把上面安装命令中的-c去掉。

4.3 安装过程中报错

解决方法:问题出在源的问题上,源文件在C:\Users\zyl\.condarc

把里面的https改成http,并把“- defaults”删除。或者可以直接用下面的源:(复制到.condarc文件里面)

channels:
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64
show_channel_urls: true

注意1:如果安装错误了,需要把整个虚拟环境都删除,然后再重新安装:

进入anaconda的控制管理器,选择环境选项--当前虚拟环境pytorch,点击左下角的remove

注意2:在安装pytorch的时候最好不要再点安装界面,这样很容易下载安装包的时候断掉(亲测多次)

5、检验是否安装成功pytorch

能够检验是否成功安装pytorch, cuda, cudnn,并能输出GPU型号

import torch

torch.cuda.is_available()

x=torch.randn(4,4)

print(x)

返回结果如上图,则说明整个过程安装完成。

### Windows 10PyTorch 安装指南 在 Windows 10 系统上安装 PyTorch 需要遵循特定的步骤来确保兼容性和性能优化。以下是详细的说明: #### 1. 环境准备 为了成功安装 PyTorch,需确认以下条件已满足: - **操作系统**: Windows 10 或更高版本。 - **Python 版本**: 推荐 Python 3.8 至 3.11[^1]。 - **CUDA 和 cuDNN (可选)**: 如果计划使用 GPU 加速,则需要安装与硬件匹配的 CUDA 和 cuDNN。 #### 2. 安装 PyTorch 可以通过两种主要方式完成 PyTorch安装:在线安装和离线安装。 ##### 2.1 在线安装 访问官方 PyTorch 官网配置工具页面(https://pytorch.org/get-started/locally/),根据系统需求选择合适的选项组合。例如,对于支持 CUDA 11.0 的环境,可以执行如下命令进行安装: ```bash conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch ``` 或者如果偏好 `pip` 工具,也可以通过以下命令实现相同目标: ```bash pip install torch==1.7.0+cu110 torchvision==0.8.0+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html ``` 上述命令适用于 CUDA 11.0 的情况;若采用最新版 CUDA 如 11.8,则应调整为: ```bash pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118 ``` 此部分提供了最新的稳定版本以及对应的依赖项[^3]。 ##### 2.2 离线安装 当网络受限时,可以选择下载本地包并手动安装。以 `torchvision`为例,其具体操作流程包括从指定链接获取压缩文件,并利用 pip 解压安装: ```bash pip install /path/to/downloaded/torchvision-package.whl ``` 注意替换路径至实际存放位置[^2]。 #### 3. 验证安装 完成以上步骤后,可通过编写简单测试脚本来验证安装是否正确无误。下面是一个基本示例用于检测 CPU/GPU 支持状况: ```python import torch print(f'Torch version: {torch.__version__}') if torch.cuda.is_available(): print('CUDA is available.') else: print('No CUDA support detected.') ``` 该脚本会打印当前使用的 PyTorch 版本号及是否存在可用的 NVIDIA 显卡资源。 ---
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值