PytorchがGPUを使っているかを確認する方法[Pytorch]
import torchprint(torch.cuda.is_available())>>Truetorch.cuda.get_device_name(0)>>'GeForce GTX 1080 Ti'
Pytorchに関する基礎知識 ...
cuDNNのバージョンを確認(Nvidia-docker2で)
# cuDNNのバージョン確認$ dpkg -l | grep "cudnn"
上のコマンドを打ってみた例
# システムにインストールされているパッケージ一覧を表示$ dpkg -l | grep "cudnn"ii libcud ...YOLOはOpenCV3.4.1に対応していないみたい[18.7.10時点]
インストールに失敗した私の環境
OpenCV: 3.4.1
Python: 3.6
CUDA: 9.1
実行した手順$ git clone cd darknet~/darknet/にある Makef ...