我正在尝试在虚拟环境中使用 conda 运行tensorflow。这些是我采取的步骤:
我使用conda create --name tf python=3.9创建了一个新的 conda 环境
我使用激活环境
conda activate tf
Run Code Online (Sandbox Code Playgroud)
我确保安装了 cuda 和 cuDNN
conda install -c conda-forge cudatoolkit=11.2.2 cudnn=8.1.0
Run Code Online (Sandbox Code Playgroud)
我配置了系统路径
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/
Run Code Online (Sandbox Code Playgroud)
我自动化了它
mkdir -p $CONDA_PREFIX/etc/conda/activate.d
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/' > $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
Run Code Online (Sandbox Code Playgroud)
我确保安装了tensorflow
pip install tensorflow==2.11.*
Run Code Online (Sandbox Code Playgroud)
然后我尝试验证安装
python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"
Run Code Online (Sandbox Code Playgroud)
这是我遇到错误的地方:
Traceback (most recent call last):
File "<string>", line 1, in <module>
File "/home/sbrent/research/Oman/venv/lib/python3.8/site-packages/tensorflow/__init__.py", line 37, in <module>
from tensorflow.python.tools import module_util as _module_util
File "/home/sbrent/research/Oman/venv/lib/python3.8/site-packages/tensorflow/python/__init__.py", line 24, in <module> …Run Code Online (Sandbox Code Playgroud)