Equim-chan / Mortal

🚀🀄️ A fast and strong AI for riichi mahjong, powered by Rust and deep reinforcement learning.
https://mortal.ekyu.moe
GNU Affero General Public License v3.0
929 stars 118 forks source link

Installation problems with pytorch #1

Closed KimamanaNeko closed 2 years ago

KimamanaNeko commented 2 years ago

You recommand installing pytorch with pip. But I found that the default installation is the cpu version of PyTorch.

I tried to test your code and got the error: AssertionError: Torch not compiled with CUDA enabled

It is recommended to specify the GPU version of PyTorch to be used in the documentation BTW I used this code to install PyTorch and fix the cuda error reporting problem pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio===0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

Equim-chan commented 2 years ago

This had already been documented, read it carefully.

[...] Check pytorch's doc on how to install pytorch in your environment. Personally, I recommand installing pytorch with pip.