技術標籤：python隨筆python

pytorch 錯誤 ImportError: numpy.core.multiarray failed to import

前言

我下載安裝 Pytorch1.7.1 GPU版，但在測試是否支援GPU時彈出了

'ImportError: numpy.core.multiarray failed to import ’

一開始我以為是numpy安裝問題，於是我解除安裝numpy,重新安裝。但結果也是彈出同樣錯誤。

最後發現是numpy版本問題。我遇到錯誤的numpy版本是1.19.4。

安裝合適的numpy版本

首先是解除安裝numpy(如果沒有就不用了)

pip uninstall numpy

由於我之前在另外一個環境安裝了pytorch,我查看了他的numpy版本是1.16.6

於是我安裝了1.16.6的numpy

pip install numpy==1.16.6

其他版本numpy我沒有嘗試，有興趣自行嘗試。也希望在評論區告知一下

執行的測試GPU程式碼

# 檢驗Pytorch是否支援GPU
import torch
a = torch.cuda.is_available()
print(a)

ngpu= 1
# Decide which device we want to run on
device = torch.device(
 
"cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3,3).cuda())

結果