cuda11.3安装+cudnn11.3安装+torch1.10.2安装+torchvision0.11.3下载(pytorch环境搭建) |
您所在的位置:网站首页 › 安装cudnn条件 › cuda11.3安装+cudnn11.3安装+torch1.10.2安装+torchvision0.11.3下载(pytorch环境搭建) |
1.查看NVIDIA显卡的CUDA适用版本 电脑主页右键点开NVIDIA控制面板 点开左下角系统信息,点开组件信息。 2.下载cuda cuda toolkit下载地址https://developer.nvidia.com/cuda-toolkit-archive 下载下来是这样的压缩包 4.创建虚拟环境 conda create -n python38 python==3.8.2 创建虚拟环境 conda activate python38 激活虚拟环境 在环境中下载 conda info --envs查看anaconda安装路径 5.下载pytorch 在官网查看对应的版本https://pytorch.org/。 pip3 install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio===0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html 官网 https://pytorch.org/get-started/previous-versions/pytorch官网 6.如果网络不好,一直加载不了,直接下载适合版本的torch和torchvision两个‘.whl’文件到本地,放入python38文件夹下。https://download.pytorch.org/whl/torch_stable.html 进入cmd命令,进入python38文件路径下 cd.. 退出目录一层 cd D: 进入D盘 cd D:\pythonEnvironment\python38 进入目录下 pip install torch7.查看是否cuda可用 输入代码 import torch print(torch.__version__) print(torch.cuda.is_available()) print(torch.version.cuda)输出为true,说明已经正确安装。 参考:https://blog.csdn.net/java_pythons/article/details/114782621 https://blog.csdn.net/weixin_43288986/article/details/106147746 |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |