有的時候一個Linux系統中很多cuda和cudnn版本，根本分不清哪是哪，這個時候我們需要進入conda的虛擬環境中，查看此虛擬環境下的cuda和cudnn版本。

在conda虛擬環境中安裝cuda和cudnn。

1. 查看torch版本

import torch print(torch.__version__) # 結果 # 1.0.0a0

2. 查看cuda版本

import torch print(torch.version.cuda) # 結果 # 9.0.176

3. 查看cudnn版本

import torch print(torch.backends.cudnn.version()) # 結果 # 7301

總結

以上是生活随笔為你收集整理的在Python中，查看torch、cuda和cudnn的版本的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。