ISSAC GYM解決RuntimeError: nvrtc: error: invalid value for --gpu-architecture (-arch)的問題

泪水下的笑靥發表於2024-12-03

起因

一開始我什麼都沒幹,就突然報了這麼一個錯誤:
image
我震驚了,因為我以為又是cuda沙雕了,沒想到網上一查,真是如此

分析

一些網上部落格的解決方法
都說其實是因為pytorch的版本問題,然後重新安裝之類的blablabla
我也照做了,但是,卻並沒有解決問題:
image
就在我非常狂躁之時,我發現我的輸出竟然還是
image
但我conda裡明明已經安裝1.13.1了??
image

哈?咋回事?
image
我又回到程式碼看看,裡面的版本確實也是這麼輸出的。
於是我在控制檯輸出了下:我以為conda裡的python環境也會是1.13.1,沒想到:
image

呃呃呃呃呃

解決

於是我在這篇部落格查到了:
這篇部落格
原來是conda 和pip的版本不匹配,我用conda安裝了1.13.1,但是我程式碼裡的

import torch

卻預設是pip的1.10.0,前者是cuda117的,後者是cuda113的,這當然不對了!!!
於是我用pip進行了解除安裝:

pip uninstall torch

然後再執行:
image

沒問題了,也正常跑起來了:
image

總結

遇到問題不要慌,慢慢解決它
最後感謝網際網路各個大佬的部落格!!

相關文章