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"。

  • 相关阅读:
    COGS 859. 数列
    9.7noip模拟试题
    hash练习
    9.6noip模拟试题
    9.5noip模拟试题
    poj 2117 Electricity
    洛谷P1993 小 K 的农场(查分约束)
    9.2noip模拟试题
    洛谷 P1273 有线电视网(dp)
    面试题收集
  • 原文地址:https://www.cnblogs.com/wf-strongteam/p/15407693.html
Copyright © 2011-2022 走看看