1 回答

TA貢獻(xiàn)1895條經(jīng)驗(yàn) 獲得超7個(gè)贊
import torch使用not驗(yàn)證安裝pytorch。下面的示例代碼,來(lái)源.
from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)
如果上述在 Jupyter Notebooks 中引發(fā)相同的問(wèn)題,并且如果您已經(jīng)啟用了 GPU,請(qǐng)嘗試重新啟動(dòng) Jupyter notebook 服務(wù)器,因?yàn)橛袝r(shí)它需要重新啟動(dòng),用戶報(bào)告。
當(dāng)我嘗試從 Jupyter notebook 導(dǎo)入這個(gè)包時(shí),我收到以下錯(cuò)誤消息:ModuleNotFoundError: No module named 'torch'。然后,我嘗試從 Anaconda 導(dǎo)航器為我的環(huán)境(手電筒)安裝 Jupyter notebook 應(yīng)用程序。重新啟動(dòng)我的 Jupyter notebook 并運(yùn)行 import torch,這次成功了
否則,您需要僅 CPU 版本的 PyTorch。
conda install pytorch torchvision cpuonly -c pytorch
添加回答
舉報(bào)