我想给企业做网站怎么做,wordpress文章生成分享图片不显示,.net网站方案,龙岩网站建设全包本方案的前提是假设机子上已经有一个版本的cuda#xff0c;现在需要支持新的torch2.1.2和torchvision0.16.2#xff0c;于是来安装新的cuda
一、选择版本
如果我想安装支持torch2.1.2的cuda版本#xff0c;到官网#xff08;https://pytorch.org/get-started/previous-ve…本方案的前提是假设机子上已经有一个版本的cuda现在需要支持新的torch2.1.2和torchvision0.16.2于是来安装新的cuda
一、选择版本
如果我想安装支持torch2.1.2的cuda版本到官网https://pytorch.org/get-started/previous-versions/#v200查看版本对应发现11.8或者12.1可以支持torch2.1.2和torchvision0.16.2 nvidia-smi查看自己的电脑能支持什么版本根据结果我们的设备最高只能运行12.0版本达不到12.1所以选择安装11.8 二、安装CUDA
到官网找到11.8https://developer.nvidia.com/cuda-toolkit-archive根据指引进行安装即运行以下命令
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
sudo sh cuda_11.8.0_520.61.05_linux.run运行run之后取消选中Driver其他默认然后Install。
【注意如果这里没有Driver可以参考其他博客先安装好Nvidia显卡驱动】 三、安装Cudnn
进入官网https://developer.nvidia.com/rdp/cudnn-archive我要安装CUDA11.8所以选择for CUDA 11.x下的内容 下载完成后运行以下命令来解压和配置
# 解压
tar -vxf ./cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz
# 复制
sudo cp cudnn-linux-x86_64-8.9.7.29_cuda11-archive/include/* /usr/local/cuda-11.8/include
sudo cp cudnn-linux-x86_64-8.9.7.29_cuda11-archive/lib/libcudnn* /usr/local/cuda-11.8/lib64
# 添加读写权限
sudo chmod ar /usr/local/cuda-11.8/include/cudnn.h
sudo chmod ar /usr/local/cuda-11.8/lib64/libcudnn*
cat /usr/local/cuda-11.8/include/cudnn_version.h | grep CUDNN_MAJOR -A 2 # 测试安装是否正常最后如下图显示则说明安装成功 四、环境配置
最后运行官网中一开始的命令来完成安装没有环境的先用前两行代码创建一个
# conda create -n torch21 python3.10 -y
# conda activate torch
conda install pytorch2.1.2 torchvision0.16.2 torchaudio2.1.2 pytorch-cuda11.8 -c pytorch -c nvidia五、其他操作切换/卸载CUDA版本
1切换cuda版本
安装的时候我们电脑下可能已经有默认的cuda版本安装新版本是是会支持覆盖的 另外也可以通过以下命令切换
sudo rm -rf /usr/local/cuda #删除之前创建的软链接
sudo ln -s /usr/local/cuda-11.8 /usr/local/cuda #创建链接到cuda-11.0的软链接
nvcc -V #查看当前cuda版本2卸载CUDA
如果想卸载CUDA进入对应路径运行以下命令后选中相应的以卸载CUDA12.0为例然后Done。
cd /usr/local/cuda-12.0/bin
sudo ./cuda-uninstaller参考链接
ubuntu配置多版本cudacudnn环境及版本切换方法