自分でいい組み合わせを見つけ次第ここにメモっていく
本当にバージョン合わせが嫌．
- PythonはConda環境で入れることを想定，CUDAとcuDNNはTSUBAMEのModuleで入れることを想定
| Python | CUDA | cuDNN | 備考 |
|---|---|---|---|
| 3.8? | 10.2.89 | 8.1 |
pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 -c pytorchで対応するtorchを入れられるrdkitがpython3.8に対応してないかも |
| 3.7 | 11.6.2 | 8.3 |
pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidiaで対応するtorchを入れられるconda install -c rdkit rdkitでrdkitもいける |