Tensorflow 2.4.1 - Couldn't invoke ptxas.exe

user_9572236 0
Viewed 5565

I try to run Tensorflow with GPU support (GTX 1660 SUPER).

I created an enviroment using anaconda, than installed cudatoolkit (version 11.0.221) and tensorflow-gpu (version 2.4.1). Afterwards, I downloaded cuDNN (version 8.0.4), and copied all files from cuDNN's bin folder to my environment's bin folder at anaconda3\envs\<env name>\Library\bin. In my script, I've set the memory limit to my GPU's memory using tf.config.experimental.set_memory_growth.

When I run the script (which uses convolutional algorithms), I get a warning that says Couldn't invoke ptxas.exe --version which comes after an Call to CreateProcess failed. Error code: 2 error. After the launch failure, I get: Relying on driver to perform ptx compilation. Modify $PATH to customize ptxas location.

I've already tried switching to cuDNN version 8.1.1.

How I fix this?

anaconda tensorflow python
2 Answers

For the benefit of community adding @Zuk Levinson comment

Solves the issue by using

tensorflow=2.3, cudnn=7.6.5 and cudatoolkit=10.1
user_142902440

I got a new fix for this.

First I tried using tensorflow=2.3, cudnn=7.6.5 and cudatoolkit=10.1 as mentioned in previous answers. However, every time I put a model to train, the process was going stale and the training seemed to be stuck in epoch 1.

I then managed to include ptxas in my conda environment by running conda install -c nvidia cuda-nvcc The packages I am using are:

tensorflow=2.9, cudnn=8.1.0, cudatoolkit=11.2.2, cuda-nvcc=11.7.99 and python=3.9

I am running everything on windows 10 flawlessly now.

user_197366100
Related