zoukankan      html  css  js  c++  java
  • Linux pytorch安装 cuda,torch,torchvision版本选择

    安装好anaconda后,安装pytorch:

    (1)使用pip install** ,代替conda install***更方便

    (2)需要用到GPU资源的话,要加cuda的内容。linux查看本机cudu版本的命令为:cat /usr/local/cuda/version.txt

    (3)知道了cuda的版本,去查看对应torch和torchvision的版本

     我的cuda版本是10.0.130,所以我要安装的pytorch版本是1.1,1.2这些。我选择1.2版本,那torch对应也是1.2版本,torchvision就选择0.4.0版本。

    (4)直接在shell里输入:pip install torch==1.2.0 torchvison==0.4.0,回车,等他运行结束

    (5)shell键入“python”进入python环境,然后输入“import torch",看看是否成功,”print(torch.cuda.is_available()) ”查看cuda版本是否对应。如果没有按照上面的版本对应的话,他会返回“False”,对应上了,就返回“True"。

  • 相关阅读:
    动态规划_leetcode70
    动态规划_leetcode64
    动态规划_leetcode63
    PHP处理base64编码字符串
    PHP解决h5页面跨域
    PHP对象转数组
    jQuery 正则
    mysql重置密码
    yii框架学习(获取插入后的id)
    nginx 之 root和alias
  • 原文地址:https://www.cnblogs.com/wf-strongteam/p/15407693.html
Copyright © 2011-2022 走看看