在学习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独立环境
(2)切换到pytorch环境
activate pytorch
(3)桌面上 右键-NVIDIA控制面板,在帮助中点击系统信息,点击组件,查看CUDA的版本信息。我的CUDA是10
第二步:
去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错误可以重新试下,或者先下载到本地,再安装
第三步:
安装完后,在python的交互环境中进行测试,若import troch,没有报错,则说明成功;torch.cuda.is_available() 返回的是True,说明GPU版本的pytorch可以使用
第四步:
配置 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
错误二:torch.cuda.is_available()返回的False
解决办法:先查看系统信息中cuda的版本号和 选择的cuda版本是否一致,若不一致,需要重新安装pytorch;或者将cuda版本进行更新,具体更新操作如下:
右键-我的电脑-管理-设备管理器,找到显示适配器,选择NVIDIA,右键-更新驱动程序,将更新到最新版本,重启电脑即可