cuda版本的pytorch包下载地址戳这里
win+R->输入cmd->输nvcc -V回车
cuda 11.0
输入以下命令来查找 CUDA 的安装路径:
Windows:
where nvcc
输入以下命令来查找 cuDNN 的版本号:
Windows:
where cudnn*
cuDNN 8.0
本机安装的是cuda 11.0,cuDNN 8.0
Python3.7,GPU版本的torch1.7.0 ,torch0.8.1
进入之后,输入conda env list,可以查看虚拟环境列表
输入activate virpytorch37gpu,进入虚拟环境
退出虚拟环境的命令:deactivate
怎么在这里手动安装已经手动下载好的包呢?
numpy1.19.2
如果anaconda prompt和navigator不好安装,也可以在pycharm环境中安装,不一定非得在一棵树上吊死。
pandas 1.0.1
matplotlib 3.1.3
opencv-python安装的是cpu版本4.2.0.32
进入anaconda prompt黑窗口,然后激活进入某虚拟环境,然后输入pip install D:\Acomputer\opencv_python-4.2.0.32-cp37-cp37m-win_amd64.whl
,输入conda命令安装,我并没有安装成功。
dataclasses0.8
typing-extensions3.7.4.3
安装pytorch gpu1.7.0版本前还需要安装上述dataclasses0.6,typing-extensions3.7.4.3
future0.18.3
这些我都是在pycharm里面的安装的
这是我选的版本,你可以根据自己需要选择适合自己的版本来安装
python选用的是3.7.6版本的
对应的dataclasses版本要选对,不然还是安装不了pytorch
dataclasses选择手动下载再来安装,下载地址在官网
python3.7 dataclasses得装0.6的,dataclasses0.7和0.8对应的python版本是3.6的。
发现torch只能安装一种类型,要么是cpu版本的,要么是gpu版本的
pip install D:\Acomputer\torch-1.7.0+cu110-cp37-cp37m-win_amd64.whl
import torch
检查是否安装了 GPU 版本
print(torch.cuda.is_available())
如果输出为 True,则表示您安装了 GPU 版本,并且可以利用 GPU 进行计算。如果输出为 False,则表示您安装的是 CPU 版本。