您的位置:首页 > 其它

cuda、pytorch 安装问题

2020-04-07 12:15 986 查看

cuda,pytorch 安装问题

基本思路是先确认自己的硬件和软件环境,比如 gpu型号、linux系统版本,gpu驱动版本。然后选择对应的 cuda和 pytorch 版本进行安装。

确认Linux内核和版本命令

#确认内核命令
uname -r

cat /proc/version

参考连接:查看内核命令

查看linux版本命令
参考连接:查看版本号

查看gpu及cuda 版本命令

gpu驱动
nvidia-smi

cuda安装及版本查看
参考连接:linux安装cuda 此连接下最后在确认 ./deviceQuery 时 通过make 命令后,如果结果还是 result=false, 尝试重启一下机器可能会成功

gpu的驱动分为运行时驱动以及正常硬件驱动。我理解的运行时驱动就是cuda, 要注意命令 nvcc 下的 编译时驱动 版本查看,是否和安装的cuda版本一致,不一致时可能会有问题, 改动方法参考连接

pytorch安装

在确定cuda环境安装好之后,pytorch安装进去官网参考相关版本及命令即可,可新建一个虚拟环境,在虚拟环境下进行操作。

  • 点赞
  • 收藏
  • 分享
  • 文章举报
不要重复造轮子 发布了2 篇原创文章 · 获赞 0 · 访问量 40 私信 关注
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: