Win 10はTensorFlow 2をインストールする.1.0+PyTorch1.3.1紀実

9149 ワード

冬休みになって、また新しい環境を振り回す時間になった.現在(2020年1月15日現在)、TensorFlowの最新バージョンは2.1.0、PyTorchの最新バージョンは1.3.1である.強迫症としても,可能な工程化を考慮し,それらを同じ環境に設置することが望ましい.
基本環境:
  • windows10
  • Python3.6

  • まず、AnacondaにcvというConda環境を新規作成します.
    conda create -n cv python=3.6
    

    インストール速度を速めるため、condaソースを清華ソースに変更します..condarcの内容は次のとおりです.
    channels:
      - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
      - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
      - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
      - defaults
    show_channel_urls: true
    

    PyTorchをインストールするには、インストール可能なバージョンを参照してください.
    conda search pytorch=1.3.1
    Loading channels: done
    # Name                       Version           Build  Channel
    pytorch                        1.3.1 cpu_py36h9f948e0_0  anaconda/pkgs/main
    pytorch                        1.3.1 cpu_py36h9f948e0_0  pkgs/main
    pytorch                        1.3.1 cpu_py37h9f948e0_0  anaconda/pkgs/main
    pytorch                        1.3.1 cpu_py37h9f948e0_0  pkgs/main
    pytorch                        1.3.1     py3.5_cpu_0  anaconda/cloud/pytorch
    pytorch                        1.3.1 py3.5_cuda101_cudnn7_0  anaconda/cloud/pytorch
    pytorch                        1.3.1 py3.5_cuda92_cudnn7_0  anaconda/cloud/pytorch
    pytorch                        1.3.1     py3.6_cpu_0  anaconda/cloud/pytorch
    pytorch                        1.3.1 py3.6_cuda101_cudnn7_0  anaconda/cloud/pytorch
    pytorch                        1.3.1 py3.6_cuda92_cudnn7_0  anaconda/cloud/pytorch
    pytorch                        1.3.1     py3.7_cpu_0  anaconda/cloud/pytorch
    pytorch                        1.3.1 py3.7_cuda101_cudnn7_0  anaconda/cloud/pytorch
    pytorch                        1.3.1 py3.7_cuda92_cudnn7_0  anaconda/cloud/pytorch
    

    PyTorch 1に気づく.3.1対応するcudaバージョンは10.1または9.2であるため、システム内のcudaバージョンはこれを下回ってはならない.PyTorch 1を取り付ける.3.1:
    conda install pytorch=1.3.1=py3.6_cuda101_cudnn7_0 torchvision cudatoolkit=10.1
    

    ネットワーク上の理由により、一部のパケットをダウンロードする際にHTTPエラーが発生する可能性があり、インストールが成功するまで上記のコマンドを繰り返します.
    次にTensorFlowをインストールします.インストール可能なバージョンを見てみましょう
    conda search tensorflow=2
    Loading channels: done
    # Name                       Version           Build  Channel
    tensorflow                     2.0.0 eigen_py36h457aea3_0  anaconda/pkgs/main
    tensorflow                     2.0.0 eigen_py36h457aea3_0  pkgs/main
    tensorflow                     2.0.0 eigen_py37hbfc5123_0  anaconda/pkgs/main
    tensorflow                     2.0.0 eigen_py37hbfc5123_0  pkgs/main
    tensorflow                     2.0.0 gpu_py36hfdd5754_0  anaconda/pkgs/main
    tensorflow                     2.0.0 gpu_py36hfdd5754_0  pkgs/main
    tensorflow                     2.0.0 gpu_py37h57d29ca_0  anaconda/pkgs/main
    tensorflow                     2.0.0 gpu_py37h57d29ca_0  pkgs/main
    tensorflow                     2.0.0 mkl_py36h781710d_0  anaconda/pkgs/main
    tensorflow                     2.0.0 mkl_py36h781710d_0  pkgs/main
    tensorflow                     2.0.0 mkl_py37he1bbcac_0  anaconda/pkgs/main
    tensorflow                     2.0.0 mkl_py37he1bbcac_0  pkgs/main
    

    condaソースのTensorFlowは2.0.0にのみ更新されているため、pipでインストールし、デフォルトで最新バージョンをインストールします.
    pip install tensorflow 
    

    インストール完了後のimport tensorflow as tfで問題が発生しました.
    ImportError: DLL load failed: The specified module could not be found.
    Githubで解決策を見つけました.https://github.com/tensorflow/tensorflow/issues/35749
    方法1:TensorFlowを2.0.0【O__O”...】方法2:ダウンロードしてインストールする:https://support.microsoft.com/en-us/help/2977003/the-latest-supported-visual-c-downloads
    親測定方法2実行可能
    最後にcudaが使えるかどうかをテストします.
    import torch
    print(torch.cuda.is_available())
    print(torch.version.cuda)
    
    import tensorflow as tf
    print(tf.test.is_gpu_available())
    

    出力:
    True
    10.1
    2020-01-15 20:52:48.968059: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll
    WARNING:tensorflow:From E:/0WorkSpace/PyCharm_Projects/Kaggle/Bengali/test.py:11: is_gpu_available (from tensorflow.python.framework.test_util) is deprecated and will be removed in a future version.
    Instructions for updating:
    Use `tf.config.list_physical_devices('GPU')` instead.
    2020-01-15 20:52:51.762533: I tensorflow/core/platform/cpu_feature_guard.cc:142] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2
    2020-01-15 20:52:51.763974: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library nvcuda.dll
    2020-01-15 20:52:51.765366: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1555] Found device 0 with properties: 
    pciBusID: 0000:01:00.0 name: GeForce RTX 2060 computeCapability: 7.5
    coreClock: 1.2GHz coreCount: 30 deviceMemorySize: 6.00GiB deviceMemoryBandwidth: 312.97GiB/s
    2020-01-15 20:52:51.765712: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll
    2020-01-15 20:52:51.765887: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cublas64_10.dll
    2020-01-15 20:52:51.766218: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cufft64_10.dll
    2020-01-15 20:52:51.766447: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library curand64_10.dll
    2020-01-15 20:52:51.804424: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusolver64_10.dll
    2020-01-15 20:52:51.804613: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusparse64_10.dll
    2020-01-15 20:52:51.804787: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudnn64_7.dll
    2020-01-15 20:52:51.805795: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1697] Adding visible gpu devices: 0
    2020-01-15 20:52:53.983713: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1096] Device interconnect StreamExecutor with strength 1 edge matrix:
    2020-01-15 20:52:53.983903: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1102]      0 
    2020-01-15 20:52:53.984016: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1115] 0:   N 
    2020-01-15 20:52:53.986073: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1241] Created TensorFlow device (/device:GPU:0 with 4604 MB memory) -> physical GPU (device: 0, name: GeForce RTX 2060, pci bus id: 0000:01:00.0, compute capability: 7.5)
    True
    

    终わって花を散らす~~~