qq_51737232的博客

原创 pytorch环境配置

即成功安装gpu版本（若安装cpu版本的pytorch则会输出false）cmd命令行输入nvidia-smi，可查看本电脑的显卡及版本。在清华源中找到与自己cuda版本对应的pytorch版本。进入虚拟环境， 若开头括号显示创建的虚拟环境名称则成功。此处推荐一个安装办法（参考自windSeS博主。需勾选环境变量，若未勾选则需手动添加。，若显示刚刚创建的虚拟环境则创建成功。打开anaconda prompt。若开头有括号（base）即安装成功。没有报错即安装成功，输入。然后在虚拟环境下输入。