检查pytorch是否安装成功、查看torch和cuda的版本 | 您所在的位置:网站首页 › ts3100怎么和电脑安装 › 检查pytorch是否安装成功、查看torch和cuda的版本 |
方式 1
在安装完成pytorch和cudatoolkit之后,需要检查一下是否安装成功,可以直接查看当前环境下安装的所有包来验证: conda list可以看到pytorch版本和cuda的版本分别是1.10.1和11.3.1 也可以在python代码中进行验证和查看 import torch # 如果pytorch安装成功即可导入 print(torch.cuda.is_available()) # 查看CUDA是否可用 print(torch.cuda.device_count()) # 查看可用的CUDA数量 print(torch.version.cuda) # 查看CUDA的版本号在命令行中的python编译器中结果如下: 还有另外一种方式查看CUDA的版本,就是在命令行中输入如下命令: nvcc -V nvcc --version
nvcc查看的就是在电脑中配置的环境变量所指向的CUDA驱动 |
CopyRight 2018-2019 实验室设备网 版权所有 |