当前位置: 首页 > news >正文

win10+python3.6+cuda9+pytorch1.1.0安装

    为了让torch可以使用显卡GPU加速,需要安装对应版本的cudatoolkit和pytorch。这里我的nvidia显卡驱动是9.1版本,只能安装cudatoolkit9。

    一般支持gpu加速的显卡大部分都是英伟达nvidia系列,都自带了nvidia驱动,所以不需要安装nvidia 驱动,我们只需要看一下驱动版本:桌面右键->nvidia控制面板->左下角系统信息->

    这里显示的是9.1,所以只能安装cudatoolkit9.0及以下版本。 

    到cudatoolkit下载页面选择合适自己系统的开发工具。

    https://developer.nvidia.com/cuda-toolkit-archive

     

    这里不管是下载network,还是local版本,最后都可以安装成功。

    正式安装cudatoolkit

    在选项这一步,有一些组件是没必要安装的,比如visual studio integration。 

    安装成功的截图:

    安装成功之后,可以在控制台查看nvcc 版本信息:    

    至此,安装已经成功了一半。表明cuda是可以被调起的,只是要让python库来调用,还需要继续安装。

    这里选择的python版本是3.6.7,因为cuda9对应的pytorch大部分都支持python3.6。

    剩下安装pytorch,torchvision等库,我是下载的whl文件,然后进行pip install的方式安装的,最早通过conda来安装,一个是anaconda会安装很多无关的库,再一个就是正式安装的时候一直在solving environment \|。无法继续安装。

    我下载的torch,torchvision的whl文件地址:

    https://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-win_amd64.whl
    https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp36-cp36m-win_amd64.whl

    

    安装的时候直接pip install torch-1.1.0-cp36-cp36m-win_amd64.whl torchvision-0.3.0-cp36-cp36m-win_amd64.whl,最后还会安装torch依赖的numpy six等库。

    pip list查看安装的库:

     python交互式命令行下验证: 

    在最后torch版本的安装中,不需要关心cudatoolkit的安装,前面已经安装成功了。只需要单独安装torch,torchvision即可。 

http://www.lryc.cn/news/26091.html

相关文章:

  • 【2023】某python语言程序设计跟学第二周内容
  • spring源码篇——BeanDefinition的注册
  • virtualbox7虚拟机中安装苹果macOS big sur系统详细教程
  • 用spectralayers 简单去一下人声做个伴奏
  • 高峰对话|深度探讨「多云与边缘」
  • 开发手册——一、编程规约_2.常量定义
  • Sandstorm 建设者亮点——2023 年 2 月
  • MyBatis快速入门
  • Mysql的一些提权方式(mysql提权、UDF)
  • 【2023】DevOps、SRE、运维开发面试宝典之Docker相关面试题
  • 圣杯布局的实现方式
  • RecastDemo用法
  • IIC总线式驱动开发(mpu6050)(二)
  • 盘点一下那些远程办公的神仙公司
  • Spring Cloud Alibaba全家桶(四)——微服务调用组件Feign
  • 安装pytorch
  • 自动化测试 Appium之Python运行环境搭建 Part2
  • LeetCode 2 - 两数相加
  • 用Python实现九九乘法表的几种方式,最简单只需一行代码
  • 入职外包三个月,我提桶跑路了
  • 企业邮箱对企业有哪些好处以及便捷性
  • ConvNeXt V2实战:使用ConvNeXt V2实现图像分类任务(一)
  • 3.2 报错整理
  • 从0开始学python -46
  • JavaScript事件委托机制详解
  • 【项目实战】MySQL中union和union all的相同点与不同点
  • ChatGPT最牛应用,让它帮你更新网站新闻吧!
  • 乌班图安装kvm并配置网络
  • 蓝库云|ERP系统在企业数字化转型中最常用的八大功能
  • Pytorch学习笔记#1:拟合函数/梯度下降