本文档详细介绍了如何在anaconda环境下安装Torch-GPU和Tensorflow-GPU。针对cuda驱动版本10.2.150，cudnn 7.6.4，推荐安装cudatoolkit 10.1和torch 1.7.1。在遇到torch.cuda.is_available()返回False的问题时，提供了重新安装pytorch-gpu的解决步骤，并给出了下载链接。对于Tensorflow-GPU的安装，提到了可能出现的错误及其解决方案，包括将dll文件复制到系统目录。
Torch-gpu
- anaconda 4.10.3（python3.9 base）
cuda：驱动版本==10.2.150
运行版本（驱动）==10.1.105 （441.22）
cudnn 7.6.4 （适用CUDA10.0，10.1，9.2，9.6）
(cudnn适用版本cuDNN Archive | NVIDIA Developer)
- 需要安装：（anaconda promote）
cudatoolkit：10.1
torch：1.7.1
python：3.6
提醒更新anaconda
conda update -n base -c defaults conda创建虚拟环境pytorch
conda create -n pytorch python=3.6官网安装pytorch（非GPU版本）
cudatoolkit选择10.2版本，其他根据代码需要的库安装。
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio
最低0.47元/天 解锁文章
328
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
