首先,一定要明白的就是:1.显卡驱动,2.CUDA, 3.cudnn 三个东西缺一不可
(要按照顺序, 依次安装哦)
windows上显卡驱动, CUDA, cudnn的安装:
显卡的购买:
电源的购买:
电源的安装:
显卡的安装:
- 直接插到卡槽里面,然后3×8pin(三个插口供电)就可以了
- 上紧螺丝钉
-
CUDA驱动的安装:
wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run #下载CUDA
chmod a+x cuda_10.2.89_440.33.01_linux.run #安装CUDA
sudo ./cuda_10.2.89_440.33.01_linux.run #安装CUDA
- (即, 下载的时候选好, 然后按照命令安装就可以了)
- 配置环境变量:(加了export那里CUDA的两个环境变量)
- 查看CUDA版本:(传说中RTX3090需要, CUDA11. 才行, 但是好像安装了一个11.的tookit问题就不大了??)
nvcc -V
-
Cudnn的安装(或者说,直接将文件移到相应的目录下)(下载东西之前要先注册)
tar xvf cudnn-10.2-linux-ppc64le-v8.0.4.30.tgz #Cudnn下载之后要解压
- 将文件移到相应的目录下:
- 把cudnn的
include/ 和 lib64/
目录下的文件拷贝到cuda的安装目录下的include/ 和 lib64
里面
sudo cp -P include/cudnn.h /usr/local/cuda-10.1/include
sudo cp -P lib64/libcudnn* /usr/local/cuda-10.1/lib64
sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*
- 当然最后pytorch的版本也需要CUDA的版本安装: