conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --remove channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --show channels
conda config --remove-key channels
torch.backends.cudnn.enabled
更新conda命令
conda update -n base -c defaults conda
conda search cudatoolkit
conda search cudatoolkit --info
conda install --use-local H:\装机软件\deeplearningsoftware\cudatoolkit-11.1.1-heb2d755_7.tar.bz2
conda search cudnn
conda search cudnn --info
conda install cudatoolkit=9.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
conda install cudnn=7.1.4 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/
Windows下使用:
conda install cudatoolkit=9.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
conda install -c conda-forge cudatoolkit=9.0
在conda环境中,可以直接用conda装cuda和cudnn,缺点是conda源中没有全部版本的cuda和cudnn,而且更新较慢。
conda search cuda
conda search cudnn
conda install cuda
conda install cudnn
如果上述方法安装很慢的话,可以在
https://repo.anaconda.com/pkgs/main/win-64/
找到自己想安装的文件URL,而后用第三方下载工具下载的合适目录中,而后在本地目录运行类似以下命令直接安装
conda install cudatoolkit-9.0-1.tar.bz2
conda install cudnn-7.6.5-cuda9.0_0.tar.bz2