pytorch的安装过程

标题实验室需要使用pytorch,故网上搜了教程来安装。

  1. 首先我们是在anaconda中安装的。
  2. 第一步是查看自己显卡对应的cuda版本。
  3. 第二步是去https://pytorch.org找到对应自己显卡cuda的版本并记住命令行 conda install pytorch1.0.0 torchvision0.2.1 cuda80 -c pytorch
  4. 第三步进入python界面输入
    import torch
    x = torch.rand(5, 3)
    print(x)
    输出一组类似于数组的数即为成功安装pytorch.
    输入print(torch.cuda.is_available())
    输出结果为true,cuda安装成功。

注意事项
在安装途中,即使我选择了对应的cuda版本,在安装完成后,torch.cuda.is_available()始终是false,后来我在使用get_device()命令时发现输出提示显卡驱动版本太低,但是笔记本只能安装相应官网上的驱动,于是网上搜到一张表,写了cuda对应的显卡驱动版本,重新下载安装,终于成功。

你可能感兴趣的:(笔记)