最近公司新配置了一台AI工作站,显卡用的刚发售的3060Ti,之前做的深度学习环境镜像中的cuda是10.1的,没法用了,就重新配置了一下环境,在这里做个记录,顺便给有需要的朋友看。
我默认你的电脑中已经安装了GTX3060Ti的显卡驱动(没有装的可以去NVIDIA官网下载安装),并且已经配置好了anaconda(python3.7版本,网上安装教程很多)。
0. 准备安装包
链接:https://pan.baidu.com/s/1cwhvZbjEwXQJk2mIM1XPGg,提取码:idpc
cuda_11.0.2_451.48_win10:CUDA11.0版本
cudnn-11.0-windows-x64-v8.0.5.39:对应CUDA11.0版本的cudnn包
torch-1.7.0+cu110-cp37-cp37m-win_amd64:对应CUDA11.0版本的torch离线安装包
torchvision-0.8.1+cu110-cp37-cp37m-win_amd64:对应CUDA11.0版本的torch离线安装包
torchaudio-0.7.0-cp37-none-win_amd64:对应CUDA11.0版本的torch离线安装包
1. 安装CUDA环境
1)首先安装cuda_11.0.2_451.48_win10,一路next即可;
2)解压cudnn-11.0-windows-x64-v8.0.5.39后可以看到一个解压出来的cuda文件夹,把里面的文件夹复制到“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0”中即可;
3)查看CUDA是否安装成功,cmd中输入:
nvcc -V
2. 安装新版pytorch
2.1 卸载老版安装包(如果之前未安装torch,则跳过)
如果之前有老版pytoch,需要先运行以下代码,把老版卸载掉。
pip uninstall torch
pip uninstall torchvision
2.2 离线安装torch包
使用pip安装的朋友应该都配置了清华源,但是清华源中目前还没有torch1.7.0中相关的windows安装包,所以需要下载官方包安装,在前面已经分享,我们直接下载安装。
2.2.1 还是使用pip安装,cmd中运行如下指令,离线安装:
pip install "你下载该文件存放路径\torch-1.7.0+cu110-cp37-cp37m-win_amd64.whl"
pip install "你下载该文件存放路径\torchvision-0.8.1+cu110-cp37-cp37m-win_amd64.whl"
pip install "你下载该文件存放路径\torchaudio-0.7.0-cp37-none-win_amd64.whl"
完,有什么不明白的可以留言讨论。