anaconda中使用pip安装pytorch(GPU)

anaconda中使用pip安装pytorch(GPU)

由于conda直接安装torch遇到了问题,在此记录一下另一种装法。

一、确认cuda与cudnn安装完成

我在以前已经安装完成了 在这仅确认一下cuda版本即可,cmd中打入:

nvcc --version

image-20220806175350107

可见是11.3版本

二、下载pytorch

pytorch官网

image-20220806175521302

找到对应的安装配置,复制链接,到conda中激活要安装的环境,直接粘贴后回车。

image-20220806180327136

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-nETJSa5c-1659781401625)(C:\Users\13587\AppData\Roaming\Typora\typora-user-images\image-20220806180354424.png)]

检查一下

import torch
torch.cuda.is_available()

返回True,安装成功!

相关conda命令:

conda create --name envname python=3.8 -y
conda activate envname
conda list          #查看环境中已经安装的库
conda env list		#查看已存在的环境

reference:

nda env list #查看已存在的环境


## reference:

[最详细的 Windows 下 PyTorch 入门深度学习环境安装与配置 CPU GPU 版...](https://www.bilibili.com/video/BV1S5411X7FY?p=27&vd_source=1422dc9d01b5aaaaa4457804c50b68ae)

你可能感兴趣的:(深度学习,pytorch,pip,python)