torch 的安装分为两个版本:Cpu 和 Gpu
CPU 版本 torch 的安装:(安装 anacode 的前提下)
1、控制台输入:
pip install torch==1.8.0
2、安装 torchvision
pip install torchvision
GPU版本torch 的安装:
1、查询 GPU 所对应的CUDA 版本号
nvidia -smi
2、CUDA 官网下载(NVIDA.DEVELOPER),推荐下载 11.3 的版本(经典!)
cp3.6=python 3.6
cu101=cuda 10.1
3、下载torch到本地
4、下载 torchvision
另:如果已经安装 GPU 版本的 torch,通过命令查询
ncvv -V