pytorch安装
本文记录之前安装pytorch的相关配置,以及简单步骤
参考文献:
https://blog.csdn.net/qq_23013309/article/details/103965619
https://blog.csdn.net/learningpawn/article/details/106531514
获得显卡驱动能支持的最高cuda版本
版本为10.2
cmd显示cuda为10.0
cudnn之前也安装了
安装pytorch1.2.0-GPU
选择对应版本,下载安装即可
测试如下:
torch和cuda安装成功