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 安装文件名