美文网首页
Win10下Pytorch安装、环境搭建、及报错的解决方案

Win10下Pytorch安装、环境搭建、及报错的解决方案

作者: luogps9 | 来源:发表于2019-06-03 00:27 被阅读0次

    在学习Pytorch的时候,对安装过程中遇到的问题进行总结。
    因为Pytorch可以启用GPU进行计算,所以需要查看显卡的版本信息。
    基本环境为:
    系统:win10
    python:3.6.5

    Part1:基础安装步骤

    第一步:
    anaconda安装完成后可以开始创建环境,这里以win10 系统为例。打开Anaconda Prompt
    (1)先找到anaconda的安装目录,Scripts目录下,运行activate.bat命令,会看到前面有个base
    利用conda create -n pytorch python=3.6 创建pytorch独立环境


    图片0.png

    (2)切换到pytorch环境
    activate pytorch
    (3)桌面上 右键-NVIDIA控制面板,在帮助中点击系统信息,点击组件,查看CUDA的版本信息。我的CUDA是10


    图片1.png 图片2.png

    第二步:
    去Pytorch官网上 https://pytorch.org/,选择自己对应的版本信息,我选择的是windows pip 3.6 cuda10,然后点击Run this Command,复制
    pip3 install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-win_amd64.whl
    pip3 install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp36-cp36m-win_amd64.whl
    若遇到HTTPERROR错误可以重新试下,或者先下载到本地,再安装

    图片3.png

    第三步:
    安装完后,在python的交互环境中进行测试,若import troch,没有报错,则说明成功;torch.cuda.is_available() 返回的是True,说明GPU版本的pytorch可以使用

    图片4.png
    图片5.png

    第四步:
    配置 Jupyter Notebook
    (1)先安装ipykernel:conda install ipykernel
    (2)加入环境:python -m ipykernel install --name pytorch --display-name "Pytorch for NLP"
    (3)切换回基础环境:activate base
    (4)创建jupyter notebook配置文件:jupyter notebook --generate-config

    part2:错误问题解决方法

    错误一:ModuleNotFoundError:No nodule named 'numpy.core._multiarray_umath'
    该错误是有Numpy的版本较低导致,只需要将Numpy升级到最新即可。
    pip3 install --upgrade numpy

    图片6.png

    错误二:torch.cuda.is_available()返回的False
    解决办法:先查看系统信息中cuda的版本号和 选择的cuda版本是否一致,若不一致,需要重新安装pytorch;或者将cuda版本进行更新,具体更新操作如下:

    右键-我的电脑-管理-设备管理器,找到显示适配器,选择NVIDIA,右键-更新驱动程序,将更新到最新版本,重启电脑即可


    图片7.png

    相关文章

      网友评论

          本文标题:Win10下Pytorch安装、环境搭建、及报错的解决方案

          本文链接:https://www.haomeiwen.com/subject/datqxctx.html