ubuntu16.04+pycharm+tensorflow配置

博客介绍了在虚拟环境中安装对应版本Tensorflow的方法,需先确保正确安装并切换Nvidia显卡驱动,配置对应版本cuda+cudnn,使用pip安装。还说明了打开PyCharm进行相关设置,以及PyCharm里缺少时手动添加环境变量、修改pycharm.sh的解决办法。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

首先保证自己的环境中已正确安装并切换至Nvidia显卡驱动,并已配置对应版本cuda+cudnn。

1.参考官网在虚拟环境中安装对应版本Tensorflow:

使用 pip 安装 TensorFlow

2.打开pycharm:File-Setting-Project:[name]-Project Interpreter

3.pycharm里面缺少libcublas.so.9.0解决办法:

手动添加环境变量:

PYTHONUNBUFFERED=1;LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64;usr/local/lib

 

修改pycharm.sh

关闭pycharm,找到pycharm目录,打开终端执行

sudo gedit bin/pycharm.sh

IFS="$(printf '\n\t')"后面按Enter空出一行,然后添加

LD_LIBRARY_PATH="/usr/local/cuda/lib64:/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64"

保存退出,重新运行pycharm,发现tensorflow可以用了

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值