conda 3090安装tensoflow-gpu报错ImportError: libcublas.so.9.0: cannot open shared object file

背景

显卡GeForce RTX 3090 驱动版本455.23.04 CUDA版本11.1

原因

没有在conda环境中安装cudatoolkit,使用安装命令

conda install cudatoolkit

附：Tensoflow2.x的完整安装

创建环境

conda create -n tensorflow-gpu python=3.8 cudatoolkit

进入环境

conda activate tensorflow-gpu

安装tensorflow

pip install tensorflow -i https://pypi.tuna.tsinghua.edu.cn/simple

测试环境

import tensorflow as tf
tf.test.is_gpu_available()