Nvidia Pytorch Usage

Created in October 10, 2023 by xyzzhangfan

2023   ·   Pytorch   Nvidia

Pulling the docker image

```bash
docker pull nvcr.io/nvidia/pytorch:23.09-py3
```

Runing the docker with

   docker run --gpus all -it --rm -v ./:/data nvcr.io/nvidia/pytorch:23.09-py3

Importing ASP to test it

   python
   from apex.contrib.sparsity import ASP


    Enjoy Reading This Article?

    Here are some more articles you might like to read next:

  • Some Usefull Commands in Cadence
  • Install NVmain from Github 2022
  • Using VCS and DVE to simulate Verilog Code
  • Docker Insufficient shared memory (shm) Error
  • Setup the working environment