Versions Compared

Key

  • This line was added.
  • This line was removed.
  • Formatting was changed.

...

Code Block
languagepy
# Make yaml file.
$ vi environment-gpu.yml
name: tf-gpu-1.15

dependencies:
  - python=3.6
  - tensorflow-gpu=1.15
  - keras=2.3.1
  - scikit-learn
  - scipy
  - numpy
  - matplotlib
  - ipython
  - jupyter
  - pillow
  - opencv
  - graphviz
  - cython
  - pip:
      - pydot-ng

# Create conda env
$ conda env create -f environment-gpu.yml

# Enter tf-gpu-1.15 env
$ conda activate tf-gpu-1.15
Note

If not able to activate a certain conda env, add this instruction into .bashrc

. ~/Miniconda3/etc/profile.d/conda.sh

Code Block
languagepy
# Check out the installed packages about cuda.
$ conda list
...
cudatoolkit               10.0.130                      0
cudnn                     7.6.5                cuda10.0_0
...
keras                     2.3.1                         0
keras-applications        1.0.8                      py_1
keras-base                2.3.1                    py36_0
keras-preprocessing       1.1.2              pyhd3eb1b0_0
...
tensorflow                1.15.0          gpu_py36h2b26d6b_0
tensorflow-base           1.15.0          gpu_py36h1afeea4_0
tensorflow-estimator      1.15.1             pyh2649769_0
tensorflow-gpu            1.15.0               h0d30ee6_0
...

...