1200字范文,内容丰富有趣,写作的好帮手!
1200字范文 > cuda cudnn tensorflow对应_Ubuntu18.04下安装Tensorflow+cuda+cudnn+pytorch

cuda cudnn tensorflow对应_Ubuntu18.04下安装Tensorflow+cuda+cudnn+pytorch

时间:2023-04-10 15:35:37

相关推荐

cuda cudnn tensorflow对应_Ubuntu18.04下安装Tensorflow+cuda+cudnn+pytorch

大体思路按照下面的链接完成

Ubuntu18.04安装TensorFlow-gpu2.0_dl_None-CSDN博客_ubuntu安装tensorflow2.0​

记录几点我自己的心得体会

1.首先说下我自己的环境

操作系统:Ubuntu18.04

cuda版本:10.1

cudnn版本:7.6.5

pytorch版本:1.4

Tensorflow版本:2.1需要前置安装:pycharm+anaconda3

2.如果在使用conda安装时,速度太慢,则可以考虑用pip的国内源安装

例如:

清华源:https://pypi.tuna./simple

阿里源:/pypi/simple/

豆瓣源:/simple/

注:新版ubuntu要使用https

使用方法:pip install xx -i https://pypi.tuna./simple

3.自己使用的cuda cudnn tensorflow的版本要兼容,需要在tensorflow官网查询

/in...

4.安装过程中要注意ubuntu的权限问题

5.在完成tensorflow-gpu安装后,需要把cudnn中的libcudnn.so.7文件拷贝到

/usr/local/lib

6.在新版本的tensorflow中,用以下代码验证tensorflow-gpu安装成功与否

tf.config.list_physical_devices('GPU')

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。