windows下Pytorch安装 --先下载对应的whl文件,再本地安装
作者RJ
1. 创建环境(anaconda)
conda create -n torch37 python=3.7
2. 选择对应的torch版本(whl文件)
https://download.pytorch.org/whl/torch_stable.html
3. 安装
3.1 查看cuda
终端输入
nvidia-smi
显示如下:
我电脑中显示的CUDA版本为10.2
选择自己需要安装的GPU版本
这里我自己选择torch(1.7.0)
如下图:
点击下载,我这里默认保存的路径为
C:\Users\rongjian\Downloads
3.2 安装
(base) C:\Users\rongjian>conda activate torch37
(torch37) C:\Users\rongjian>cd C:\Users\rongjian\Downloads
(torch37) C:\Users\rongjian\Downloads>pip install "torch-1.7.0+cu110-cp37-cp37m-win_amd64.whl