- 博客(4)
- 收藏
- 关注
RSS订阅原创 ImportError: libcudnn.so.6: cannot open shared object file 解决方法
在运行tensorflow-gpu时遇到了这个问题，可以通过以下命令解决。export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64在执行命令之前可以先cd到目录下，看文件是不是在里面。...
2018-03-17 13:58:43
1895
1
原创 tensorflow指定GPU与动态分配GPU memory
在tensorflow中，默认指定占用所有的GPU，如需指定占用的GPU，可以在命令行中：export CUDA_VISIBLE_DEVICES=1这样便是只占用1号GPU，通过命令nvidia-smi可以查看各个GPU的使用情况。另外，也可以在python程序中指定GPU，并且动态分配memory，代码如下import os import sys os.environ['CUDA_VISIBLE...
2018-03-15 11:02:50
4027
原创 python3.5 pip3 安装pytorch和torchvision
pytorch框架现在运用广泛，下面在python3.5中安装一下。pip3 install http://download.pytorch.org/whl/cu80/torch-0.1.11.post5-cp35-cp35m-linux_x86_64.whl pip3 install torchvision如果需要配置代理，在语句的后面加上--proxy [代理地址] 就可以了。...
2018-03-15 10:57:18
13598
原创 用pip批量安装包
在配置集群的时候，会遇到用pip安装大量包的情况，下面是用pip批量安装包的方法：首先，在已经配置好的一台机器上，将需要的包导出到requirements.txtpip freeze > requirements.txt查看requirements.txt，内容如下然后，在集群的其他机器上执行pip install -r requirements.txt就可以完成批量安装包的工作了。...
2018-03-14 14:18:14
7888
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人