Which will give an output similar to this: You can check which NVIDIA-driver you have installed by running On your system supports the selected CUDA version. Here, you only need to make sure that the NVIDIA driver CUDA and cuDNN will be installed via the prebuild binaries inīoth pip or conda. That is basically the only thing you need to In order to run pytorch on your system you would need a working NVIDIAĭriver on your machine. Now, for FastSurfer we used pytorch v1.2.0 gpu version for CUDA 10.0, which is the default for pip (see instructions for v1.2.0 (Wheel): ): ECC | | Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. | GPU Name Persistence-M | Bus-Id Disp.A | Volatile Uncorr.
0 Comments
Leave a Reply. |
AuthorWrite something about yourself. No need to be fancy, just an overview. ArchivesCategories |