Linux上tensorflow环境搭建

sudo apt install python3-pip
pip install tensorflow

pip install tf-nightly

sudo sh cuda_11.0.2_450.51.05_linux.run{{/code}}
 

CUDA 11.3下载地址：


{{code language="none"}}
wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run

sudo ln -s /usr/local/cuda-11.0/lib64/libcusolver.so.10.5.0.218 /usr/local/cuda-11.0/lib64/libcusolver.so.11