有的時(shí)候一個(gè)Linux系統(tǒng)中很多cuda和cudnn版本，根本分不清哪是哪，這個(gè)時(shí)候我們需要進(jìn)入conda的虛擬環(huán)境中，查看此虛擬環(huán)境下的cuda和cudnn版本。

在conda虛擬環(huán)境中安裝cuda和cudnn。

1. 查看torch版本

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

2. 查看cuda版本

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

3. 查看cudnn版本

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

總結(jié)

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

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