在虚拟环境中安装完cuda和cudnn想要测试是否安装成功,不能使用nvcc -V命令测试,需要在虚拟环境中安装pytorch包进行测试
#在虚拟环境中先进入python环境
python
#引入pytorch
import torch
#检查虚拟环境中cuda cudnn版本号
print(torch.version.cuda)
print(torch.backends.cudnn.version())
在虚拟环境中安装完cuda和cudnn想要测试是否安装成功,不能使用nvcc -V命令测试,需要在虚拟环境中安装pytorch包进行测试
#在虚拟环境中先进入python环境
python
#引入pytorch
import torch
#检查虚拟环境中cuda cudnn版本号
print(torch.version.cuda)
print(torch.backends.cudnn.version())