原创 deepaiDL 实用AI客栈 2022-04-26 22:48 收录于合集 #windows1 #深度学习9 #人工智能10 #机器学习7 #计算机视觉2 对新手而言,安装深度学习框架通常需要大费周折,有没有省事简便的安装方法呢 如果我们想在windows系统上安装GPU版本的AI框架,比如GPU版本的tesnorflow,通常我们会看到类似下面的安装教程 官方版本
咋看上去好像不是很复杂,但是其中坑多到你怀疑人生。
极简版本
上面我们可以看到,这里我们使用了一个叫做cudatoolkit的包,相当于在conda的虚拟环境里面自动安装了CUDA。大多数时候cudatoolkit中的CUDA和cuDNN是一起安装的,这就省去了很多麻烦,比如说AI框架和CUDA的版本对应问题以及进行手动设置环境变量等等。 具体的安装脚本可参考如下命令: 安装tensorflowconda create -n py36 python=3.6 conda config --add channels https://mirrors.tuna./anaconda/pkgs/free/conda config --set show_channel_urls yes conda install cudatoolkit tensorflow-gpu 安装pytorchconda create -n py36 python=3.6 conda config --add channels https://mirrors.tuna./anaconda/pkgs/free/conda config --set show_channel_urls yes conda install pytorch torchvision cudatoolkit -c pytorch 大家有没有觉得安装环境变得很简单了呢,欢迎评论区留言。 1 END 1 分享人工智能领域最新资讯,讲解各类实用的算法和模型。 20篇原创内容 公众号 确定 |
|