Mune's Weblog

鹿児島在住のstray radiologistです
仕事も趣味も画像関係…

2020.07.06

2020年07月06日 | 仕事関係
週休5日ではありません!
久々に環境構築を行いました。今回は、筐体を開けてグラフィックボード(なんとTITAN RTX!!!)換装から。
環境構築のメインはPytorchの導入。
今まではtensorflow+kerasで行っていたのですが、
1. TensorflowがTF2に変わり、また勉強しなおさざるを得なくなってしまったこと
2. NiftyNetがMONAIとして生まれ変わり、tensorflow+kerasからPytorchへ変わったこと
3. 参加させてもらうことになったAIベンチャー企業がPytorchをお好みのこと
4. あのPreferred NetworksがChainerの開発を止め、Pytorchへ合流したこと
これらの理由で踏み切りました。まだdefine by runがよくわかってませんが...



今回はシンプルなアドバイスをいただいていたので、大きなトラブルなく環境構築できたっぽいです。
手順としては
1. GPUドライバーのインストール
2. Anaconda最新版の導入 (Python 3.7だったので、Python 3.6の仮想環境を用意)
3. Visual Studio 2015 Communityの導入
4. CUDA Toolkitのインストール (CUDA10.2がOKhttps://pytorch.org/
5. 仮想環境をactivateして、Pytorchのインストール コマンドプロンプトから conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
6. この状態で、一応CUDAは使えるみたいだけど (print(torch.cuda.is_avialable()) >> Trueなので)
 cuDNNもDLして、CUDA toolkitのフォルダにコピー

さて、お次はMONAIの導入だ。

最新の画像もっと見る

コメントを投稿