- 安装CUDA和CUDNN
访问https://developer.nvidia.com/cuda-gpus查询自己的gpu是否在网站的列表内,如果在则访问https://developer.nvidia.com/cuda-downloads下载自己显卡驱动所对应的cuda版本,安装。
访问https://developer.nvidia.com/cudnn获取cudnn,将cudnn内的文件拷贝到cuda安装目录下的同名目录下即可。
2.虚拟环境配置 - 访问https://www.anaconda.com/获取anaconda以安装。
- 打开anaconda prompt,执行命令 conda create -n tf python=3.7.4,其中tf就是环境名,新建完环境以后激活环境使用conda activate tf。
图 1 anaconda新建虚拟环境 - 激活虚拟环境后执行如下命令,安装所需要的包
pip install numpy1.16.5
pip install tensorflow-gpu1.13.1
pip install keras2.1.5
pip install opencv-python4.2.0.32
pip install pyside25.14.2
pip install matplotlib3.1.1 - 打开pycharm,选择工程,点击File->Setting->Project interpreter 选择刚搭建好的虚拟环境。
cudnn cuda 安装
最新推荐文章于 2024-03-18 14:53:00 发布