接上一篇:笔记本显卡1660Ti,Ubuntu18.04安装Cuda、Cudnn、Anaconda、PyTorch、Tensorflow和Keras

上一篇链接: 笔记本显卡1660Ti,Ubuntu18.04安装Cuda、Cudnn、Anaconda、PyTorch、Tensorflow和Keras_炖大鹅乀的博客-CSDN博客

五、安装Pytorch

5.1 根据cuda的版本,选择Pytorch。官网链接:Start Locally | PyTorch 

接上一篇:笔记本显卡1660Ti,Ubuntu18.04安装Cuda、Cudnn、Anaconda、PyTorch、Tensorflow和Keras_第1张图片

 也可以点击上面桔黄色的install previous version of PyTorch,选择历史版本,由于当时安装的时间比较早,选择的还是11.3的版本,所以需要在历史版本里面找(11.6的没试过,有兴趣的可以试一下)。

接上一篇:笔记本显卡1660Ti,Ubuntu18.04安装Cuda、Cudnn、Anaconda、PyTorch、Tensorflow和Keras_第2张图片

5.2 创建Pytorch虚拟环境,安装Pytorch。在Anaconda基础环境激活的条件下,创建Pytorch虚拟环境并激活,输入上图所示 #CUDA 11.3 下的安装命令,等待安装完成。(注:在创建虚拟环境安装Pytorch的时候由于Anaconda需要搜索相关的安装包,所以等待的时间比较长。此外,这两个步骤都需要手动输入yes)

conda create -n torch python=3.9.12
conda activate torch
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch

5.3 测试。在torch环境下,输入代码,结果如下。

python
import torch
torch.__version__

接上一篇:笔记本显卡1660Ti,Ubuntu18.04安装Cuda、Cudnn、Anaconda、PyTorch、Tensorflow和Keras_第3张图片

六、安装Tensorflow及Keras

6.1 在Anaconda中创建并激活Tensorflow虚拟环境。

conda create -n tf2 python=3.8
conda activate tf2

6.2 安装Tensorflow2.6.0和Keras2.6.0。(注:Tensorflow可以选择cpu和gpu两个版本,这里直接选择安装gpu版本)

conda install tensorflow-gpu==2.6.0
conda install keras==2.6.0

至此,全部过程讲述完毕,感谢支持,如有不足和疑问欢迎评论。

你可能感兴趣的:(pytorch,深度学习,tensorflow,keras,ubuntu)