打开APP
userphoto
未登录

开通VIP,畅享免费电子书等14项超值服

开通VIP
Win10下Pytorch安装、环境搭建、及报错的解决方案
在学习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 installhttps://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-win_amd64.whl
pip3 installhttps://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
本站仅提供存储服务,所有内容均由用户发布,如发现有害或侵权内容,请点击举报
打开APP,阅读全文并永久保存 查看更多类似文章
猜你喜欢
类似文章
Windows下用anaconda装python,添加jupyter的kernel以及装包
深度学习环境配置指南!(Windows、Mac、Ubuntu全讲解)
从硬件配置、软件安装到基准测试,1700美元深度学习机器构建指南
超详细配置教程:用Windows电脑训练深度学习模型
Caffe学习系列(13):数据可视化环境(python接口)配置
Win10上安装Keras 和 TensorFlow(GPU版本)
更多类似文章 >>
生活服务
热点新闻
分享 收藏 导长图 关注 下载文章
绑定账号成功
后续可登录账号畅享VIP特权!
如果VIP功能使用有故障,
可点击这里联系客服!

联系客服