一、問題

今天跑一個(gè)項(xiàng)目代碼遇到了一個(gè)問題，大致是pytorch和CUDA版本不匹配，然后我就降低pytorch版本，結(jié)果1.7中網(wǎng)絡(luò)不包含我所需的函數(shù)，我直接pip install pytorch==1.8，這樣產(chǎn)生了錯(cuò)誤：Torch not compiled with CUDA enabled。

二、解決方法

nvcc -V

檢查一下當(dāng)前CUDA版本

根據(jù)自己的CUDA版本下載安裝whl文件

參考鏈接：https://download.pytorch.org/whl/torch_stable.html

分別

pip install torch-1.8.1+cu101-cp38-cp38-linux_x86_64.whl pip install torchcsprng-0.2.1+cu101-cp38-cp38-linux_x86_64.whl pip install torchvision-0.9.1+cu101-cp38-cp38-linux_x86_64.whl

總結(jié)

以上是生活随笔為你收集整理的Pytorch错误：Torch not compiled with CUDA enabled的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。