conda安装tensorflow和pytorch

之前装tensorflow时候,看网上的教程基本上都是先去nvidia官网下载对应NVIDIA驱动版本的cuda以及它的更新包和对应的cudnn,然后一个包一个包的安装,按照教程走也基本没问题,最大的困难就是网速了,不过根据个人经验,一般早上下载服务器在国外的资源,速度会相对较快,但还是麻烦,对新手来说,不够友好。

后来发现可以使用conda安装tensorflow,它会自动帮你解决依赖项问题,不用自己再另外安装cuda和cudnn以及其他依赖包,非常简单粗暴,我很喜欢,哈哈。

最近又想装下pytorch,也想使用conda一键安装,但是遇到了一个意外,死活下不下来pytorch官网的包,最终发现pytorch官网上的那句conda安装pytorch命令,已经指定了pytorch包的下载源是官网的服务器,难怪一直下不下来,然后我就自己添加了清华镜像源,发现两分钟就下好了。所以把这个记下来,顺便分享给大家。

下面是具体步骤:

1,添加conda的清华镜像源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ 
conda config --set show_channel_urls yes

2,安装tensorflow

conda install tensorflow-gpu==1.11 #如果不指定版本,默认下载最新版

3,安装pytorch

conda install pytorch torchvision cuda80 -c pytorch # 这是官网给的命令

改成清华镜像源

conda install pytorch torchvision cuda80 -c channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

也可以不指定channel,也会使用清华源。

很简单吧。

验证tensorflow装是否可以正常使用

import tensorflow as tf

x = tf.constant{[1.0])

sess = tf.Session()

print(sess.run(x))

验证pytorch安装是否可以正常使用

import torch

x = torch.Tensor([1.0])
print(x)

y = x.cuda()
print(y)

from torch.backends import cudnn
print(cudnn.is_acceptable(y)) # 输入为True

如果上面这段代码没有提示错误信息,那么恭喜你,可以开始使用pytorch了。

    原文作者:阿贵
    原文地址: https://zhuanlan.zhihu.com/p/52498335
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞