- 我的电脑是 windows10系统,在任务管理器中 查看是否有GPU
2.去官网 按照对应的命令下载即可,
https://pytorch.org/get-started/locally/
CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台
我是这么选择的,
这里有一个关于Windows(64bit)下安装pytorch-gpu
https://www.huaweicloud.com/articles/7c2f84bc1bc86dec1c7f30561469ec64.html 的链接,我也不知道问什么要安装那么多东东。我没装,因为我输入 nvcc -V 啥也不是
3.激活虚拟环境,开启安装,
pip3 install torch1.9.0+cu102 torchvision0.10.0+cu102 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
pip install -i https://pypi.douban.com/simple ipykernel 为了将 该pytorch 的虚拟环境添加到jupyter中,
(pytorch_nlp) D:\all_env\pytorch_nlp\Scripts>python -m ipykernel install --user --name=pytorch_nlp
然后可以查看一下 安装了哪些虚拟环境kernel ,
- 启动jupyter ,切换环境
- 将官网中的代码复制,运行一下,是运行在GPU上的。。
6.以上就是pytorch GPU的安装,