各種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()
    
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章