栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

【pytorch安装教程】win10+anaconda安装pytorch

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

【pytorch安装教程】win10+anaconda安装pytorch

    安装NVIDIA驱动
    下载地址
    点击开始搜索,自动搜索自己电脑显卡的型号,产品型号记得选择带notebooks的;
    点击立即下载,安装包下载完,按默认配置安装即可。
    安装Anaconda
    下载地址
    可能需要先注册一个账号,安装包下载完,按默认配置安装即可。
    进入 Anaconda prompt 命令窗口
    先使用清华源,必须使用http
    conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
    conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
    conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
    conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
    conda config --set show_channel_urls yes
    
    打开C:Users你的用户名 文件夹下修改.condarc文档,将其中的内容替换为如下代码:
    channels:
      - defaults
    show_channel_urls: true
    default_channels:
      - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
      - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
      - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
    custom_channels:
      conda-forge: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      msys2: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      bioconda: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      menpo: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      pytorch: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      simpleitk: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
    
    回到 Anaconda prompt 命令窗口
    // PyTorch是名字,可以改,必须指定python版本号
    conda create -n PyTorch python=3.9
    // 查看 Anaconda 的所有环境
    conda info --envs
    
    进入pytorch官网
    官网地址

    选择相应配置,复制配置下面的命令回到 Anaconda prompt 命令窗口
    输入下面的命令进入PyTorch环境
    conda activate PyTorch
    

    输入刚刚复制的命令,等待下载完成
    conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
    
    测试安装成功
    重新进入 Anaconda prompt 命令窗口
    // 进入python环境
    python
    // 测试
    import torch
    torch.cuda.is_available()
    
    输出为
    True
    
    至此安装成功
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/701129.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号