ubuntu18.04安装tensorflow+pytorch笔记

最新推荐文章于 2024-06-17 16:04:49 发布

一往无前的洋仔

最新推荐文章于 2024-06-17 16:04:49 发布

阅读量223

点赞数

分类专栏： Ubuntu

本文链接：https://blog.csdn.net/weixin_45295666/article/details/108013265

版权

Ubuntu 专栏收录该内容

9 篇文章 3 订阅

订阅专栏

记录多次安装的点点滴滴

安装tensorflow、pytorch、CUDA和cuDNN方式一：

conda install tensorflow-gpu==1.14 cudatoolkit cudnn
conda install pytorch==1.4.0 torchvision==0.5.0 -c pytorch

-c制定下载频道(或者网址）
conda会自行配置CUDA和cuDNN版本

验证安装成功与否

>>>python
>>>import tensorflow as tf
>>>tf.test.gpu_device_name()
>>>import torch
>>>import torchvision
>>>print(torch.cuda.is_available())	#输出为true即可
>>>torch.__version__

至于CUDA的验证和cuDNN的验证自行搜索
补充：为不同tensorflow或pytorch版本创建的conda虚拟环境依赖不同的CUDA和cuDNN，因此可以在主机上装一个高版本的显卡驱动，而各虚拟环境的CUDA和cuDNN版本只需低于驱动版本即可（后续会验证该做法），不解可自行搜索“cuda与driver驱动之间的对应关系”。