安装的各块内容联系
步骤
安装Anaconda
下载链接: https://www.anaconda.com/
可以更换路径:(不要有中文路径)
都打钩
利用conda安装pytorch
从开始菜单中打开Anaconda Prompt
输入命令:
conda env list
查看Anaconda中有哪些虚拟环境:
然后创建一个名字叫Pytorch的虚拟环境:
输入命令:
conda create -n pytorch python=3.9
自此装好;
激活虚拟环境:
输入命令:
conda activate pytorch
查看刚刚安装的虚拟环境里面有啥包:
输入命令:
conda list
如果以后要移除虚拟环境:
输入命令:
conda remove -n pytorch --all
注意:上面的pytorch是安装的时候的虚拟环境的名称!!!
安装CUDA
三个东西一定要版本相互匹配;
第一步确定自己显卡的算力:
先看看自己显卡的版本
确定显卡算力:
打开链接:
https://en.wikipedia.org/wiki/CUDA
算力是7.5
确定CUDA Runtime版本:
10.0-12.3都可以;
然后看看自己的驱动CUDA Driver Version
输入命令:
nvidia-smi
版本为11.7
最终确定我们可以安装的CUDA Runtime 版本可以是10.0、10.2、11.0、11.1-11.7.1;
推荐使用右面的;
下载安装最新的驱动程序
打开nvidia官网:
链接:https://www.nvidia.com/en-us/
点Drivers:
输入自己电脑的类型:
下载安装即可!
更新之后:
那实际上,更新以后CUDA Runtime版本可以安装最新的12.3的;
打开pytorch官网:查看:
在pytorch环境中安装:
复制官网中的安装命令:
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
安装即可!
安装完后看看pytorch虚拟环境中的包:
验证Pytorch:
自此成功!!!