1、查看pytorch版本
第一步,查看所有环境信息:conda info --envs
第二步,找到安装有pytorch的环境并激活:source activate torch_env
第三步,进入python运行环境:python (ctrl d 退出python运行环境)
第四步,查看torch版本:import torch 换行 print(torch.version)

2、查看cuda版本:nvidia-smi

3、cuda对应的pytorch:https://pytorch.org/get-started/previous-versions/

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐