Pytorch安装 -- 先下载对应的whl文件,再本地安装。(100%成功)

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

猜你喜欢

转载自blog.csdn.net/laplacez/article/details/126685595
今日推荐