Added functionality to support specifying quantity of a resource in call to show-gpus

[Saikrishna-Achalla]

This PR fixes issue #1915. Previously, users were not able to specify the GPU quantity that they wanted. This PR allows users to do so by running sky show-gpus V100:4 to find the cloud providers and regions that offer 4 V100 GPUs.

The output is now as follows:

sky show-gpus V100:8

GPU   QTY  CLOUD   INSTANCE_TYPE  DEVICE_MEM  vCPUs  HOST_MEM  HOURLY_PRICE  HOURLY_SPOT_PRICE  REGION            
V100  8    AWS     p3.16xlarge    16GB        64     488GB     $ 24.480      $ 7.344            ap-northeast-1    
V100  8    GCP     (attachable)   -           -      -         $ 19.840      $ 5.920            us-central1       
V100  8    Lambda  gpu_8x_v100    16GB        92     448GB     $ 4.400       -                  asia-northeast-1  

GPU        QTY  CLOUD  INSTANCE_TYPE       DEVICE_MEM  vCPUs  HOST_MEM  HOURLY_PRICE  HOURLY_SPOT_PRICE  REGION          
V100-32GB  8    AWS    p3dn.24xlarge       32GB        96     768GB     $ 31.212      $ 9.364            ap-northeast-1  
V100-32GB  8    Azure  Standard_ND40rs_v2  -           40     672GB     $ 22.032      $ 3.530            eastus

Failure output is as follows:

sky show-gpus V100:5

Resource 'V100' with quantity 5 not found. Try 'sky show-gpus --all' to show available accelerators and their quantities.

Tested (run the relevant ones):

• Any manual or new tests for this PR (please specify below)
• All smoke tests: pytest tests/test_smoke.py
• Relevant individual smoke tests: pytest tests/test_smoke.py::test_fill_in_the_name
• Backward compatibility tests: bash tests/backward_comaptibility_tests.sh

[romilbhardwaj]

Thanks @Saikrishna-Achalla! This is a great first step to implementing this feature. Left some comments.

I got a runaway Resource 'V100' with quantity 4 not found. Try 'sky show-gpus --all' to show available accelerators and their quantities. when I ran the command below. Is this expected?

(base) ➜  ~ sky show-gpus V100:4
*NOTE*: for most GCP accelerators, INSTANCE_TYPE == (attachable) means the host VM's cost is not included.

GPU   QTY  CLOUD  INSTANCE_TYPE       DEVICE_MEM  vCPUs  HOST_MEM  HOURLY_PRICE  HOURLY_SPOT_PRICE  REGION
V100  4    AWS    p3.8xlarge          16GB        32     244GB     $ 12.240      $ 3.672            ap-northeast-1
V100  4    Azure  Standard_NC24rs_v3  -           24     448GB     $ 13.460      $ 4.506            centralus
V100  4    Azure  Standard_NC24s_v3   -           24     448GB     $ 12.240      $ 4.097            centralus
V100  4    GCP    (attachable)        -           -      -         $ 9.920       $ 2.960            us-central1     Resource 'V100' with quantity 4 not found. Try 'sky show-gpus --all' to show available accelerators and their quantities.

[romilbhardwaj]

Thanks @Saikrishna-Achalla! Left some comments.

[romilbhardwaj]

Great work @Saikrishna-Achalla! Left some comments.

[romilbhardwaj]

One final nit, otherwise good to go! Thanks for this contribution @Saikrishna-Achalla! 🚀