安裝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/
- pytorch官網:
- 安裝torchvision
https://pypi.org/project/torchvision/
- 測試
import torch torch.__version__ torch.cuda.is_available()