Similarly, you could install the CPU version of pytorch when CUDA is not installed. This environment variable is useful for downstream installations, such as when pip installing a copy of pytorch that was compiled for the correct CUDA version. # Determine CUDA version using /usr/local/cuda/version.txt fileĬUDA_VERSION=$(cat /usr/local/cuda/version.txt | sed 's/.* \(\ \.\ \).*/\1/') # Determine CUDA version using /usr/local/cuda/bin/nvcc binaryĬUDA_VERSION=$(/usr/local/cuda/bin/nvcc -version | sed -n 's/^.*release \(\ \.\ \).*$/\1/p') Įlif then # Determine CUDA version using default nvcc binaryĬUDA_VERSION=$(nvcc -version | sed -n 's/^.*release \(\ \.\ \).*$/\1/p') Įlif /usr/local/cuda/bin/nvcc -version 2
0 Comments
Leave a Reply. |