测试CUDA
查看系统CUDA版本,方法:
- cat /usr/local/cuda/version.txt
- nvcc -V
具体见ubuntu16.04查看CUDA和cuDNN版本与Ubuntu14.04 查看安装好的CUDA
安装PyTorch
https://pytorch.org/get-started/previous-versions/
补充:cuda10.0 安装pytorch
-
pytorch 1.0稳定版正式发布之前,使用下面的命令安装pytorch,版本是pytorch 0.4.1
conda install pytorch torchvision -c pytorch
-
2018.12.7在 NeurIPS 大会上,Facebook 正式发布 PyTorch 1.0 稳定版。新的安装命令如下
conda install pytorch torchvision cuda100 -c pytorch