Windows10でCUDA Toolkit, PyTorch, Python(Anaconda)でGPUでディープラーニングできるように環境設定

ブログ

参考にしたサイト

NVIDIA グラフィックスドライバ,NVIDIA CUDA ツールキット 11.0,NVIDIA cuDNN 8.0.5 のインストール(Windows 上)

インストールしたもの

CUDA Toolkit

CUDA Toolkit 11.0 Update 1 Downloads

Anaconda

Anaconda Installers

これ↓の 64-Bit Graphical Installer (457 MB) と書いてあるところをクリックして、流れに従ってダウンロード&インストールする。

Anaconda Installers

PyTorch 1.7.1

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

上のコマンドはここINSTALLING PREVIOUS VERSIONS OF PYTORCHの中にある。CUDA 11.0と書いてあればよい。自分の使用したいCUDAのバージョンに対応したものを使う。

結果

Python 3.7.9 (default, Aug 31 2020, 17:10:11) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type “help”, “copyright”, “credits” or “license” for more information.

>> import torch
>> torch.version
>> torch.version.cuda
'11.0'
>> x = torch.rand(5, 3)
>>x
tensor([[0.2836, 0.2533, 0.6447],
[0.5962, 0.9144, 0.4351],
[0.8404, 0.7213, 0.3332],
[0.5363, 0.6731, 0.5580],
[0.0970, 0.1784, 0.0115]])
>> x.cuda()
tensor([[0.2836, 0.2533, 0.6447],
[0.5962, 0.9144, 0.4351],
[0.8404, 0.7213, 0.3332],
[0.5363, 0.6731, 0.5580],
[0.0970, 0.1784, 0.0115]], device='cuda:0')

うまく動いた!

タイトルとURLをコピーしました