各种python包的安装方法及常见错误

安装numpy

  • 如果已经安装了numpy,需要升级numpy时,可能会遇到numpy无法卸载的问题,这可能是由于anaconda prompt没有以管理员身份运行。

  • import其他包,可能会遇到如下错误,这是也可能是由于numpy包的问题,尝试升级numpy解决

    RuntimeError: module compiled against API version 0xb but this version of numpy is 0xa

安装pytorch

  • 前提:已经安装anaconda,最好为pytorch创建一个新的python环境。
  • 检查机器是否满足安装CUDA的要求,并查看本机的显卡型号(设备管理器 -> 显示适配器)
    https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html#install-cuda-software
  • 更新显卡驱动
    https://www.nvidia.cn/Download/index.aspx?lang=cn
  • 安装CUDA Toolkit
    https://developer.nvidia.com/cuda-downloads
  • 检查CUDA安装
    cmd中输入命令:nvcc -V 查看CUDA版本
  • 配置cuDNN
    • 下载地址:https://developer.nvidia.com/cudnn,下载前需要先注册并登录,cuDNN版本要和CUDA版本匹配。
    • 下载后解压,会生成cuda/include、cuda/lib、cuda/bin三个目录。
    • 分别将cuda/include、cuda/lib、cuda/bin三个目录中的内容拷贝到CUDA安装路径下对应的include、lib、bin目录下。
    • 将bin所在的目录添加到系统环境变量 PATH 中。
  • 安装pytorch
    • pytorch官网:https://pytorch.org/
    • 联网条件下,可以直接在anaconda prompt中运行下面命令:conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
    • 没网时,可以在清华镜像中下载对应版本,同样要注意和python、CUDA版本匹配:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
  • 安装torchvision
    https://pypi.org/project/torchvision/
  • 测试
    import torch
    torch.__version__
    torch.cuda.is_available()
    

你可能感兴趣的:(python)