1.查看cuda最大支持版本号:
进入电脑控制面板,右上角帮助,查看系统信息,选择组件可以查看到电脑所支持的最高cuda版本信息
2.下载Miniconda
在官网,选择需要的版本下载(对应python版本),直接运行,点击安装即可,建议安装到自定义目录,方便查找。可以通过在cmd输入conda检查安装情况。
3.通过cuda安装pytorch
创建一个python3.7的虚拟环境,name是环境名字
conda create -n name python=3.7
conda activate name
激活该环境
conda activate name
激活环境后到Pytorch官网在命令行安装套件
安装完套件后,如果后续还缺某些包,直接在该虚拟环境激活状态下pip或者conda安装即可
conda deactivate # 退出虚拟环境