pytorch(GPU)环境配置
笔记本windows10安装cuda11.4、安装cuDNN11.4、安装pytorch-gpu(1.11.0+cu113:2022.3最新版只适配到cuda11.3,所以安的也是这个)
最近想在笔记本上跑程序,调用一下笔记本上的显卡(gtx 1650),方便跑一些小规模代码。
清华镜像:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
创建环境:
conda create --name pytg python=3.8
附一些常见conda指令,以下几行无需执行
删除环境(我不会告诉你第一遍我失败了)
conda remove -n pytg --all
开启环境
conda activate pytg
关闭环境
conda deactivate
网址:https://developer.nvidia.com/cuda-toolkit-archive
等安装完了,可以在cmd中测试一下
nvcc -V
V要大写,如果有如下显示,则安装成功
环境变量
关于环境变量,我安装完之后检查了一下,我的环境变量已经是配好的,但是网上有的攻略说需要自己配置一下,但是我的看我的已经配好了就没管。
下载下来是一个压缩文件,解压一下,把这里面的这几个文件夹(如下图)全部粘贴到cuda的安装目录下,覆盖粘贴即可
默认cuda路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
在python中测试一下。代码比较简单,所以直接用cmd里输入python,进入python编译器执行就行。
import torch
import torchvision
print (torch.cuda.is_available())
如果不报错,且返回true那就成功了
如果下载失败,会有很多半不拉拉的包,加上以前攒了也不少,都清理了吧
conda clean -a
https://blog.csdn.net/weixin_43667077/article/details/104518764
https://blog.csdn.net/qq_42999949/article/details/122774246
https://blog.csdn.net/weixin_54546190/article/details/122097856
https://blog.csdn.net/thy0000/article/details/122878599
https://sleepingbug.blog.csdn.net/article/details/77370456