diff --git a/README.md b/README.md index e3d741b1..62671964 100644 --- a/README.md +++ b/README.md @@ -183,8 +183,27 @@ virtualenv venv ./venv/Scripts/activate.bat pip install -r requirements.txt pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 - ``` +Then, create a file that named 'test.py' + +```test.py +import torch +x = torch.rand(5, 3) +print(x) +print(torch.cuda.is_available()) +``` +Run the script via and confirm the output looks like this (the first part should have just random numbers, but the last line must read ; if it does not, CUDA is not installed properly) + +```terminal +tensor([[0.2141, 0.7808, 0.9298], + [0.3107, 0.8569, 0.9562], + [0.2878, 0.7515, 0.5547], + [0.5007, 0.6904, 0.4136], + [0.2443, 0.4158, 0.4245]]) +True +``` + +If the last word is "True" that means you installed it successfully. ## Docker Image