首先在終端中輸入python進(jìn)入python交互式環(huán)境
import torch
print(torch.__version__) #注意是雙下劃線
官網(wǎng):https://pytorch.org/get-started/previous-versions/
CUDA Toolkit版本及可用PyTorch對(duì)應(yīng)關(guān)系總結(jié)(參考官網(wǎng))文章來源:http://www.zghlxwxcb.cn/news/detail-739169.html
cuda版本確定后,pytorch要依賴此cuda版本,因?yàn)閜ytroch安裝跟cuda對(duì)應(yīng)的,比如下圖,torch1.11.0只適配cuda10.2、11.3;不適配cuda10.1等其他版本cuda。文章來源地址http://www.zghlxwxcb.cn/news/detail-739169.html
到了這里,關(guān)于關(guān)于python pytorch 與CUDA版本相關(guān)問題的文章就介紹完了。如果您還想了解更多內(nèi)容,請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!