wi-lab
/

lwm

Wireless-Communication

Zero-Shot Classification

Feature-Extraction

Inference Endpoints

Model card Files Files and versions Community

Sadjad Alikhani commited on Sep 26

Commit

2261009

•

1 Parent(s): 457207f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -51,7 +51,7 @@ Once the environment is activated, install the necessary packages.
 #### **Install CUDA-enabled PyTorch**
-Change the `pytorch-cuda` version based on your system requirements.
 ```bash
 conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

 #### **Install CUDA-enabled PyTorch**
+While inference runs efficiently on CPU, you may require a GPU for training downstream tasks. Follow the instructions below to install CUDA-enabled PyTorch. Be sure to adjust the `pytorch-cuda` version according to your system's specifications.
 ```bash
 conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia