windows下conda安装pytorch_gpu和cupy

1.安装pytorch_gpu
查看cuda版本：cmd中输入nvcc -V
我的cuda版本：11.1

官方文档的重要性：
链接: https://pytorch.org/get-started/previous-versions/.

找到对应cuda版本的conda命令。

在conda prompt新建环境中输入：

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

一个小技巧：尽量将所有用到的库包一行写完，conda会帮助找到不冲突的版本安装上各个库包的。
上面这行代码末尾的conda-forge为下载源，因为挂了梯子所以没有改，没有梯子怎么改下载源自行搜索。

可以看到pytorch的build为py3.6_cuda11.1_cudnn8_0，如果是cpu则无法使用cuda！！！

pycharm中新建一个try.py，复制粘贴以下内容测试一下：

import torch
print(torch.randn((1,2,1)))
print(torch.cuda.is_available())

显示True：

说明pytorch_gpu可以使用了。

2.安装cupy
cupy安装官方文档: https://docs.cupy.dev/en/stable/install.html.

pip install cupy-cuda111

111为版本号，对照自己电脑上的版本做修改。

over.