一、Python环境
- 详见之前文章Python版本管理-Miniconda
- 创建一个名为pytorch的Python3.8环境并激活
conda create -n pytorch python=3.8 conda activate pytorch
二、确认最高支持的CUDA版本
- 在cmd中执行nvidia-smi查看
三、Pytorch安装
- 打开pytorch的官网,根据自己系统选择对应的配置。最后在激活的pytorch环境中执行红色框的命令等待安装完成即可
conda create -n pytorch python=3.8
conda activate pytorch