
tensorflow
unsterbliche
这个作者很懒,什么都没留下…
展开
-
Ubuntu tensorflow2.X 调用cuda10.2
已经安装了CUDA 10.2,然后才发现TensorFlow 2.2只支持CUDA 10.1,CUDA10.2没有对应的tensorflow版本。采取下面的办法建一个软链接即可。 注意,我的cuda安装在conda虚拟环境里,不使用根目录里安装的cuda ln -s ~/anaconda3/envs/pytorch/lib/libcudart.so.10.2.89 ~/anaconda3/envs/pytorch/lib/libcudart.so.10.1 相关:conda创建虚拟环境 和原创 2022-04-11 10:19:57 · 1004 阅读 · 0 评论 -
多显卡服务器,显存明明够用,却报错 Failed to create session.
今天在服务器3上跑了下tensorflow的模型, 一直报错Failed to create session. 通常这种情况是显存不够,但是服务器3上两块1080Ti,第一块占满了,第二块空的,因此不应该没有显存. 把第二块显卡'1'设置为可见 import os os.environ['CUDA_VISIBLE_DEVICES'] ='0,1' 依旧报错 把显卡都设置为不可见 os...原创 2019-04-15 15:28:21 · 1708 阅读 · 0 评论 -
源码安装deepchem
先下载安装 git clone https://github.com/deepchem/deepchem.git # Clone deepchem source code from GitHub cd deepchem gpu=1 bash scripts/install_deepchem_conda.sh deepchem # If you want GPU s...原创 2019-06-29 15:03:28 · 1594 阅读 · 2 评论