pycharmcuda安装
时间: 2023-11-12 16:06:26 浏览: 43
在安装Pycharm和Python之后,需要安装CUDA Toolkit 8.0和cudnn。安装cudnn的步骤如下:进入解压后的cuda文件夹,打开终端,输入以下命令:
sudo cp include/cudnn.h /usr/local/cuda-10.0/include/
sudo cp lib64/libcudnn.so.7.6.5 /usr/local/suda-10.0/lib64
然后进入一个文件并进行编辑,具体内容如下:
sudo vim /etc/ld.so.conf.d/cuda /usr/local/cuda-10.0/lib64。
需要注意的是,目前Pycharm安装的TensorFlow-gpu版本为1.4.0,测试目前只支持到CUDA Toolkit 8.0、cuDNN v6.0,虽然官网已经有CUDA Toolkit 9.0。