ubuntu18.04 cuda10.1 pt1.5.1 cudnn8.0.5
程序员文章站
2022-05-27 14:42:30
...
1.cuda 10.1(显卡驱动:450,使用的是ubuntu18.04中自带 附加程序,设定的)
https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal
nano ~/.bashrc
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export PATH=$PATH:/usr/local/cuda-10.1/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1
source ~/.bashrc
nvcc --version
2.cudnn 8.0.5
https://developer.nvidia.com/rdp/cudnn-archive
2-1根据实际软件包名称,执行下列命令
sudo dpkg -i libcudnn8_8.1.0.77-1+cuda11.2_amd64.deb
sudo dpkg -i libcudnn8-dev_8.1.0.77-1+cuda11.2_amd64.deb
sudo dpkg -i libcudnn8-samples_8.1.0.77-1+cuda11.2_amd64.deb
2-2 确认
cd /usr/src/cudnn_samples_v8/mnistCUDNN/
make clean && make
./mnistCUDNN
若出现下面内容,代表ok
Result of classification: 1 3 5
Test passed!
3.pytorch
https://download.pytorch.org/whl/torch_stable.html
对应cuda 10.1
pytorch1.5.1 - torchvision 0.6.1
推荐阅读
-
Ubuntu18.04安装CUDA10.1
-
如何在Ubuntu18.04下安装CUDA10.1和cudnn
-
Ubuntu18.04安装NVIDIA显卡驱动 cuda10.1 cudnn7.6.1
-
Ubuntu18.04安装CUDA10.1
-
Ubuntu18.04安装NVIDIA显卡驱动、cuda10.1以及cudnn-10.1-7.6.4
-
Ubuntu18.04安装CUDA10.1和cuDNN v7.6.5
-
ubuntu18.04 cuda10.1 pt1.5.1 cudnn8.0.5
-
ubuntu18.04卸载cuda10.1,安装cuda10.0又双踩坑
-
ubuntu18.04安装和卸载cuda11.0 cuda10.1 ,遇到的各种问题
-
利用预训练好的模型实现 FlowNetPytorch(ubuntu18.04 cuda10.1 cudnn7.6.4)