下载Anaconda
然后自己配置环境变量:
接下来,进入NVIDIA控制面板,点击帮助-系统信息:
可以查看到驱动程序版本。
这里安装 CUDA 11.6 。
安装CUDA
下载安装包后启动,
测试,终端输入 nvcc -V ,返回CUDA版本号,即安装成功。
安装cuDNN
将压缩包解压,然后将其中的三个文件夹加入到cuda的安装目录下,
这里是 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6
直接粘贴即可。
***现在的系统环境变量:
在系统变量中:
安装
pytorch
根据对应的型号版本来选择:
更多的型号版本选择:
在里面找到对应的版本安装命令:
# CUDA 11.6
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.6 -c pytorch -c conda-forge
注意,在输入上面的安装指令前, 还是先创建一个新环境:
conda create -n name python==3.10.9
现在是创建conda环境名为pytorch:
输入 conda info --envs 查看现有环境:
切换到创建的这个环境,
然后输入安装pytorch命令:
-------------------------------------------------------------------------------------------------------------------------------
这里也给出了conda更新方法:
==> WARNING: A newer version of conda exists. <==
current version: 23.1.0
latest version: 23.3.1
Please update conda by running
$ conda update -n base -c defaults conda
Or to minimize the number of packages updated during conda update use
conda install conda=23.3.1
使用 conda update -n base -c defaults conda 就行。
-------------------------------------------------------------------------------------------------------------------------------
测试
最后进行演示:
至此,pytorch安装完成。