从网上看其他大佬的博客学来的,之前一直没成功,还以为是电脑的问题,今天偶然间发现是因为错了一个字母。
查看cuda和cudnn版本:
cat /usr/local/cuda/version.txt
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
查看当前已有环境:
conda env list
新建一个自己的环境:
conda create -n tensorflow1.9 python=3.6
source activate tensorflow1.9
conda install tensorflow-gpu==1.9
安装其他需要的包:
conda install scipy
conda install h5py
退出该环境:
source deactivate
不喜欢该环境了,把它删掉:
conda remove -n tensorflow1.9 --all
复制环境:
conda create --name new_env_name --clone tensorflow1.9
查看当前环境中有哪些包
conda list
win:
conda env list
conda create -n tensorflow1.9 python=3.6
conda activate tensorflow1.9
conda install tensorflow-gpu==1.9
deactivate
j进入环境的时候如果报错:
就把
conda activate 环境名称
换成:
conda.bat activate 环境名称
pytorch cpu安装:
conda总报错,用pip:
pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
网友评论