在安装GPU版pytorch,tensorflow的时候,往往需要先安装好NVIDIA相关软件,本人在参考大量网络资源后历经坎坷后终于成功安装好GPU版pytorch,下面简单梳理安装时的思路,以防后来者踩坑。
一、更新nvidia显卡驱动
更新驱动参考博客:
备注:驱动更新与cuda的安装是两个独立的过程,更新驱动是为了提高版本号,这样才能使用高版本的cuda更新完驱动后在命令行输入:nvidia-smi查看版本号及对应cuda版本,
注意这里的cuda版本与我们单独安装的cuda版本不一定非得一样,我们只要单独安装的cuda版本小于等于它即可。最后记得配置环境变量。同时,CUDA版本可以有多个。
二、下载安装CUDA
参考博客
备注:我选择的是10.2版本,这里的版本要与安装pytorch时的版本一致,一路确定即可,不必自定义。命令行输入:nvcc -V ,看到版本号表明安装成功。