anaconda中使用pip安装pytorch(GPU)
由于conda直接安装torch遇到了问题,在此记录一下另一种装法。
一、确认cuda与cudnn安装完成
我在以前已经安装完成了 在这仅确认一下cuda版本即可,cmd中打入:
nvcc --version
可见是11.3版本
二、下载pytorch
找到对应的安装配置,复制链接,到conda中激活要安装的环境,直接粘贴后回车。
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(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