打开 cmd 面板,输入 nvidia-smi 命令, 显示 NVIDIA 显卡基本信息和相关进程占用显存情况
算力及显卡对照表:CUDA GPUs - Compute Capability | NVIDIA Developer
GeForce GT 730-4G (DDR5) 的算力为3.5,根据算力匹配pytorch版本
配置步骤:
1.下载并安装 Anaconda;
2.在 Windows 上打开开始菜单并打开 Anaconda 命令提示符(Anaconda Prompt)。
3.创建pytorch环境;
conda create -n PyTorch python=3.7
4.安装 pytorch:
conda install pytorch==1.2.0 torchvision==0.4.0 cudat