记得将显卡的驱动升级到最新
本文针对的CUDA Version为11.7版本的安装教程
由于CUDA版本可以向下兼容,我们安装的官网的CUDA11.3的版本
Anaconda安装包http://xn--anaconda-nh9mg71epx7i/
一步一步安装即可
PyCharm Community Edition 2022.1.2https://download.jetbrains.com.cn/python/pycharm-community-2022.1.2.exe
CUDA11.7安装地址https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local提前下载好以上文件安装包
安装完Anaconda后查看环境变量(我的电脑右键属性→高级系统设置→环境变量)
需要有这四个环境变量,如果发现没有的话可以手动添加
D:\ProgramData\Anaconda3
D:\ProgramData\Anaconda3\Scripts
D:\ProgramData\Anaconda3\Library\mingw-w64\bin
D:\ProgramData\Anaconda3\Library\bin
输入指令
conda list
conda --version
找到.condarc文件(一般在打开命令提示薄的路径中或者点击C盘→用户→点击自己的用户名)
然后用记事本打开输入
channels:
- defaults
show_channel_urls: true
default_channels:
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
conda-forge: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
msys2: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
bioconda: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
menpo: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
pytorch: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
simpleitk: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
耐心等待即可,这一步都很慢
推荐使用自定义一是可以修改安装地方,默认是安装在C盘
二是如果第一次安装的话会有很多CUDA包没有
只勾选CUDA
下一步即可,我已经安装完成就不继续演示了
打开自己CUDA安装目录C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\bin
是否有如下文件
nvcc -V
pytorchhttps://pytorch.org/get-started/locally/
输入指令
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
慢慢等待即可
pip list
输入以下代码出现True
import torch
torch.cuda.is_available()