cuda和cudnn的安装
安装之前要先安装cuda和cudnn。我安装的是cuda10.1。cuda用安装包安装,都选默认就行。
相应版本的cudnn在NVIDIA官网下载。cudnn下载下来是一个zip压缩包,解压之后,把里面的文件复制到cuda安装目录的相应位置。

在命令行里输入 nvcc -V 测试是否安装成功:

Python
此处选用python3.7。python安装完毕后,修改pip镜像源为豆瓣,具体过程可以参考这里
安装tensorflow
pip install tensorflow-gpu
如果网速较慢,可以先用迅雷从这里把安装文件下载下来,在文件下载目录执行如下命令:
pip install 安装文件名
安装pytorch
从pytorch官方网站根据自己的情况获取相应的下载命令:

不出意外的话会因为网络连接不稳定而中断。可以先用迅雷将安装文件下载到本地,安装文件的下载地址会显示在命令行里,如:

然后使用pip安装:
pip install 安装文件名
本文详细介绍了CUDA10.1和cuDNN的安装步骤,包括如何选择正确的cuDNN版本,从NVIDIA官网下载并解压,以及复制文件到CUDA安装目录。同时,文章还提供了使用Python3.7和pip镜像源加速安装tensorflow和pytorch的过程。
2万+

被折叠的 条评论
为什么被折叠?



