Совпадений не найдено: tensorflow[and-cuda]Python

Программы на Python
Гость
Совпадений не найдено: tensorflow[and-cuda]

Сообщение Гость »


I am trying to install TensorFlow on my WSL on my laptop. The general installation of

Код: Выделить всё

pip install tensorflow
its successful. But it does detect GPU on my laptop. When I test GPU on it

Код: Выделить всё

python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
it does not detect any GPU even though I do have CUDA installed on the machine.

Код: Выделить всё

2024-03-09 10:15:18.941101: I tensorflow/core/platform/cpu_feature_guard.cc:210] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations. To enable the following instructions: AVX2 FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags. 2024-03-09 10:15:19.496900: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT 2024-03-09 10:15:20.199286: I external/local_xla/xla/stream_executor/cuda/cuda_executor.cc:984] could not open file to read NUMA node: /sys/bus/pci/devices/0000:01:00.0/numa_node Your kernel may have been built without NUMA support. 2024-03-09 10:15:20.199478: W tensorflow/core/common_runtime/gpu/gpu_device.cc:2251] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform. Skipping registering GPU devices... [] 
So i tried following the documentation and tried to install

Код: Выделить всё

python3 -m pip install tensorflow[and-cuda]
, but i get

Код: Выделить всё

zsh: no matches found: tensorflow[and-cuda] 
I do have CUDA installed here are some details with

Код: Выделить всё

nvidia-smi

Код: Выделить всё

+-----------------------------------------------------------------------------------------+ | NVIDIA-SMI 550.54.10              Driver Version: 551.61         CUDA Version: 12.4     | |-----------------------------------------+------------------------+----------------------+ | GPU  Name                 Persistence-M | Bus-Id          Disp.A | Volatile Uncorr. ECC | | Fan  Temp   Perf          Pwr:Usage/Cap |           Memory-Usage | GPU-Util  Compute M. | |                                         |                        |               MIG M. | |=========================================+========================+======================| |   0  NVIDIA GeForce RTX 3080 ...    On  |   00000000:01:00.0 Off |                  N/A | | N/A   58C    P8             11W /  150W |       0MiB /  16384MiB |      0%      Default | |                                         |                        |                  N/A | +-----------------------------------------+------------------------+----------------------+                                                                                           +-----------------------------------------------------------------------------------------+ | Processes:                                                                              | |  GPU   GI   CI        PID   Type   Process name                              GPU Memory | |        ID   ID                                                               Usage      | |=========================================================================================| |  No running processes found                                                             | +-----------------------------------------------------------------------------------------+ 
More details about the setup

Код: Выделить всё

Python 3.12.2 pip 24.0 conda 24.1.2 WSL 2  fedoraremix :   NAME="Fedora Remix for WSL"   VERSION="38" Microsoft Windows :Version 22H2 (OS Build 19045.4046) 


Источник: https://stackoverflow.com/questions/781 ... owand-cuda

Вернуться в «Python»