1. 创建虚拟环境
具体查看我之前写的
《在Windows中利用Python的venv和virtualenv创建虚拟环境》
2. 检查显卡版本和CUDA
这种情况是需要电脑上有单独的英伟达的显卡、或者英伟达的显卡和集显这两种情况都是可以的。)
3. 下载链接
进入PyTorch官网选择合适自己设备的下载链接
4. 下载
进入cmd 输入上一步获得的下载链接
5. 等待
6. 检测
下载完成后,进入python环境检测torch是否可用
7.下载轮文件
轮文件官网
按CTRL+F 搜索对应版本的文件
使用Motrix进行下载轮文件(只是因为torch太大采用下载器下载,其他可直接使用浏览器下载),版本对应如下
8.下载Torchvision
github pytorch和torchvision版本对应官网
我的对应版本为